diff --git a/src/communication.py b/src/communication.py
index c436e90..1b9bd42 100644
--- a/src/communication.py
+++ b/src/communication.py
@@ -191,7 +191,7 @@ class ClusterCommunicationModule():
         for conn in self.worker_conns:
             conn.close()
 
-    def run_container(self, image_name, train, train_args={})
+    def run_container(self, image_name, train, train_args={}):
         '''
         train_args
             - index