diff --git a/test/server_test.py b/test/server_test.py
index 79d412aaa1..7b0814768e 100644
--- a/test/server_test.py
+++ b/test/server_test.py
@@ -160,12 +160,12 @@ def test_404(self):
 
     @skipOnTravis('https://travis-ci.org/spotify/luigi/jobs/72953884')
     def test_save_state(self):
-        self.sch.add_task('X', 'B', deps=('A',))
-        self.sch.add_task('X', 'A')
-        self.assertEqual(self.sch.get_work('X')['task_id'], 'A')
+        self.sch.add_task(worker='X', task_id='B', deps=('A',))
+        self.sch.add_task(worker='X', task_id='A')
+        self.assertEqual(self.sch.get_work(worker='X')['task_id'], 'A')
         self.stop_server()
         self.start_server()
-        work = self.sch.get_work('X')['running_tasks'][0]
+        work = self.sch.get_work(worker='X')['running_tasks'][0]
         self.assertEqual(work['task_id'], 'A')