diff --git a/api/tests/chain/test_upscale_outpaint.py b/api/tests/chain/test_upscale_outpaint.py
index 0e524014..261a8d45 100644
--- a/api/tests/chain/test_upscale_outpaint.py
+++ b/api/tests/chain/test_upscale_outpaint.py
@@ -28,9 +28,7 @@ class UpscaleOutpaintStageTests(unittest.TestCase):
                 3,
                 0.1,
             ),
-            ServerContext(
-                # model_path="../models",
-            ),
+            ServerContext(),
             None,
             ImageParams(
                 "../models/stable-diffusion-onnx-v1-inpainting",
diff --git a/api/tests/chain/test_upscale_swinir.py b/api/tests/chain/test_upscale_swinir.py
index e2bf69fe..dfa9676e 100644
--- a/api/tests/chain/test_upscale_swinir.py
+++ b/api/tests/chain/test_upscale_swinir.py
@@ -28,9 +28,7 @@ class UpscaleSwinIRStageTests(unittest.TestCase):
                 3,
                 0.1,
             ),
-            ServerContext(
-                # model_path="../models",
-            ),
+            ServerContext(),
             None,
             None,
             sources,
diff --git a/api/tests/worker/test_worker.py b/api/tests/worker/test_worker.py
index 553094ad..6365fac9 100644
--- a/api/tests/worker/test_worker.py
+++ b/api/tests/worker/test_worker.py
@@ -18,7 +18,7 @@ from tests.helpers import test_device
 
 
 def main_memory(_worker):
-    raise Exception(MEMORY_ERRORS[0])
+    raise MemoryError(MEMORY_ERRORS[0])
 
 
 def main_retry(_worker):