1 files changed, 6 insertions, 1 deletions
diff --git a/rpc/rpc.py b/rpc/rpc.py
index 4b634a9..010715e 100644
--- a/rpc/rpc.py
+++ b/rpc/rpc.py
@@ -3,6 +3,7 @@ import os
 import zerorpc
 import gevent
 import time
+import sys
 from io import BytesIO
 import base64
 from PIL import Image
@@ -34,12 +35,16 @@ class CortexRPC(object):
     val = None
     try:
       val = self.relay.connected('rpc')
-      atexit.register(lambda: self.relay.disconnecting())
+      atexit.register(self.disconnecting)
       print('RPC connected...'.format(val))
     except zerorpc.TimeoutExpired:
       print('RPC timeout :(')
     return val
 
+  def disconnecting(self):
+    print('Disconnecting from relay', flush=True, file=sys.stderr)
+    self.relay.disconnecting()
+
   def ping(self):
     return 'pong'