diff --git a/src/demo/demos.py b/src/demo/demos.py
index 3b0bdc7..2371f1d 100644
--- a/src/demo/demos.py
+++ b/src/demo/demos.py
@@ -387,6 +387,7 @@ def check_classification_task(string):
             log.info(f"Loaded the model from the local path sucessfully!")
     except:
         log.error("No model found with the given name!")
+        return
 
     if S3_bool:
         data_path = (