diff --git a/src/device/tests/test_unitary_morpheus.py b/src/device/tests/test_unitary_morpheus.py
index b909dddb91e46751e8f0324c08474e9beea938fb..518cea9ed8894c0c14280cbcc8047961c164cda0 100644
--- a/src/device/tests/test_unitary_morpheus.py
+++ b/src/device/tests/test_unitary_morpheus.py
@@ -71,7 +71,7 @@ def test_set_config(driver: MorpheusApiDriver):
     assert results[0] is True, "Expected a succesfull result"
 
 def test_retrieve_state(driver: MorpheusApiDriver):
-    results = driver.SubscribeState()
+    results = driver.SubscribeState([])
 
     assert all(isinstance(result, bool) and result for result in results), \
             f"Subscription error: {results}"
@@ -83,7 +83,7 @@ def test_retrieve_state(driver: MorpheusApiDriver):
 
     print_data("State", state)
 
-    results = driver.UnsubscribeState()
+    results = driver.UnsubscribeState([])
 
     assert all(isinstance(result, bool) and result for result in results), \
             f"Unsubscription error: {results}"