diff --git a/model_server/extensions/ilastik/tests/test_ilastik.py b/model_server/extensions/ilastik/tests/test_ilastik.py
index 1907459d2d5d3a87be5da49e291fe15e4e1b7ca0..420cd0a07616b8c9d449520d06d5d7b6956cc7f2 100644
--- a/model_server/extensions/ilastik/tests/test_ilastik.py
+++ b/model_server/extensions/ilastik/tests/test_ilastik.py
@@ -195,7 +195,7 @@ class TestIlastikPixelClassification(unittest.TestCase):
             channel=0,
         )
         self.assertTrue(result.success)
-        self.assertGreater(result.timer_results['inference'], 1.0)
+        self.assertGreater(result.timer_results['inference'], 0.1)
 
 class TestIlastikOverApi(TestServerBaseClass):
 
diff --git a/tests/test_api.py b/tests/test_api.py
index 9cb56c685ec5cf1378cce144e498fd37ea85fb5a..6e336a9c0e46fddf1f627667618ff13516a58fa6 100644
--- a/tests/test_api.py
+++ b/tests/test_api.py
@@ -15,7 +15,7 @@ class TestServerBaseClass(unittest.TestCase):
     def setUp(self) -> None:
         import uvicorn
         host = '127.0.0.1'
-        port = 5001 # TODO: defer to run_server script and conf file
+        port = 5001
 
         self.server_process = Process(
             target=uvicorn.run,