diff --git a/test/test_class1_presentation_neural_network.py b/test/test_class1_presentation_neural_network.py
index 2d615b89500c1cc4c70c8050d39c644a1100bee3..91799922d5e807fb836a6f700ab65f23ca13292e 100644
--- a/test/test_class1_presentation_neural_network.py
+++ b/test/test_class1_presentation_neural_network.py
@@ -360,4 +360,3 @@ def test_real_data_multiallelic_refinement(max_epochs=10):
     print(refine_allele, "final motif lysine-1 rate: ", final_motif.loc[1, "K"])
 
     assert_greater(final_motif.loc[1, "K"], original_motif.loc[1, "K"])
-    
\ No newline at end of file
diff --git a/test/test_class1_presentation_predictor.py b/test/test_class1_presentation_predictor.py
index 939de5c8e6b666b73cd97e0691ab3c1bc68a0f09..9b18f5d044bc8df76e0f5befd47fab997eef5413 100644
--- a/test/test_class1_presentation_predictor.py
+++ b/test/test_class1_presentation_predictor.py
@@ -24,7 +24,7 @@ AFFINITY_PREDICTOR = None
 def setup():
     global AFFINITY_PREDICTOR
     startup()
-    PAN_ALLELE_PREDICTOR_NO_MASS_SPEC = Class1AffinityPredictor.load(
+    AFFINITY_PREDICTOR = Class1AffinityPredictor.load(
         get_path("models_class1_pan", "models.no_mass_spec"),
         optimization_level=0,
         max_models=1)
@@ -32,7 +32,7 @@ def setup():
 
 def teardown():
     global AFFINITY_PREDICTOR
-    PAN_ALLELE_PREDICTOR_NO_MASS_SPEC = None
+    AFFINITY_PREDICTOR = None
     cleanup()