diff --git a/tests/test_fed_get.py b/tests/test_fed_get.py
index 714ef7c..f49fc6a 100644
--- a/tests/test_fed_get.py
+++ b/tests/test_fed_get.py
@@ -55,7 +55,7 @@ def run(party):
     fed.init(cluster=cluster, party=party)
 
     epochs = 3
-    alice_model = MyModel.remote("alice", 2)
+    alice_model = MyModel.party("alice").remote("alice", 2)
     bob_model = MyModel.party("bob").remote("bob", 4)
 
     all_mean_weights = []