diff --git a/tests/test_api.py b/tests/test_api.py
index 3cde9af5b41559f1517342167abfa075a8a7b190..c81c2d02713558deea00a498afdec60785d5236c 100644
--- a/tests/test_api.py
+++ b/tests/test_api.py
@@ -19,8 +19,8 @@ class TestServerBaseClass(unittest.TestCase):
             daemon=True
         )
         self.uri = f'http://{host}:{port}/'
-        requests.get(self.uri + 'restart')
         self.server_process.start()
+        requests.get(self.uri + 'restart')
 
     def copy_input_file_to_server(self):
         from shutil import copyfile
@@ -65,7 +65,7 @@ class TestApiFromAutomatedClient(TestServerBaseClass):
         resp_load = requests.put(
             self.uri + f'models/dummy/load',
         )
-        model_id = resp_load.json()['model_id']
+        model_id = list(resp_load.json()['model_id'].keys())[0]
         self.assertEqual(resp_load.status_code, 200, resp_load.json())
         resp_list = requests.get(self.uri + 'models')
         self.assertEqual(resp_list.status_code, 200)