diff --git a/MTBO_assignment_test.py b/MTBO_assignment_test.py
index f4ea388..3f6f8c1 100644
--- a/MTBO_assignment_test.py
+++ b/MTBO_assignment_test.py
@@ -14,11 +14,15 @@
 
 @pytest.fixture(scope="session")
 def get_namespace():
-    script_fname = "MTBO_assignment_ans.py"
+    script_fname = "MTBO_assignment.py"
     script_content = open(script_fname).read()
 
     namespace = {}
     exec(script_content, namespace)
+    required_vars = ["ax_client", "uniformity_A", "uniformity_B", "best_reactor", "temp_diff", "pressure_diff", "num_iterations_single", "higher_or_lower"]
+    missing_vars = [var for var in required_vars if var not in namespace]
+    if missing_vars:
+        pytest.skip(f"Assignment incomplete. Missing variables: {missing_vars}")
     return namespace