diff --git a/module-3/classic-example/run_training_job.py b/module-3/classic-example/run_training_job.py
index bd3318f..668f490 100644
--- a/module-3/classic-example/run_training_job.py
+++ b/module-3/classic-example/run_training_job.py
@@ -24,7 +24,7 @@ def run_classic_example():
     train(config_path=Path("/app/conf/example.json"))
     upload_to_registry(model_name="modal-classic-example", model_path=Path("results"))
     load_from_registry(
-        model_name="modal-classic-example", model_path=Path("loaded-model")
+        model_name="modal-classic-example:latest", model_path=Path("loaded-model")
     )
     run_inference_on_dataframe(
         df_path=Path("/tmp/data/test.csv"),
diff --git a/module-3/generative-example/run_training_job.py b/module-3/generative-example/run_training_job.py
index 5e307d3..96a8d19 100644
--- a/module-3/generative-example/run_training_job.py
+++ b/module-3/generative-example/run_training_job.py
@@ -22,9 +22,9 @@ def run_generative_example():
 
     load_sql_data(path_to_save=Path("/tmp/data"), subsample=0.01)
     train(config_path=Path("/app/conf/example-modal.json"))
-    # upload_to_registry(model_name="modal_generative_example", model_path=Path("/tmp/phi-3-mini-lora-text2sql"))
-    # load_from_registry(model_name="modal_generative_example", model_path=Path("/tmp/loaded-model"))
-    run_evaluate_on_json(json_path=Path("/tmp/data/test.json"), model_load_path=Path("/tmp/phi-3-mini-lora-text2sql"), result_path=Path("/tmp/data/results.json"))
+    upload_to_registry(model_name="modal_generative_example", model_path=Path("/tmp/phi-3-mini-lora-text2sql"))
+    load_from_registry(model_name="modal_generative_example:latest", model_path=Path("/tmp/loaded-model"))
+    run_evaluate_on_json(json_path=Path("/tmp/data/test.json"), model_load_path=Path("/tmp/loaded-model"), result_path=Path("/tmp/data/results.json"))
 
 
 def main():