fix module

1 files changed, 2 insertions, 4 deletions

diff --git a/rpc/rpc.py b/rpc/rpc.py
index d64bb7e..31fedf8 100644
--- a/rpc/rpc.py
+++ b/rpc/rpc.py
@@ -23,11 +23,8 @@ class CortexRPC(object):
     print('Connecting to RPC on port {}'.format(os.getenv('RPC_PORT')))
     self.rpc = zerorpc.Server(self)
     self.rpc.bind('tcp://127.0.0.1:' + os.getenv('RPC_PORT'))
-    gevent.spawn(self.connected)
-    gevent.spawn(lambda: ready_fn(self))
     gevent.spawn(self.rpc.run)
 
-  def connected(self):
     print('RPC running on port {}'.format(os.getenv('RPC_PORT')))
     print('Connecting to Relay on port {}'.format(os.getenv('RELAY_PORT')))
     self.relay = zerorpc.Client()
@@ -39,7 +36,8 @@ class CortexRPC(object):
       print('RPC connected...'.format(val))
     except zerorpc.TimeoutExpired:
       print('RPC timeout :(')
-    return val
+    ready_fn(self)
+    # return val
 
   def disconnecting(self):
     print('Disconnecting from relay', flush=True, file=sys.stderr)