diff --git a/ait/core/server/client.py b/ait/core/server/client.py
index 60718d2d..bfd6297c 100644
--- a/ait/core/server/client.py
+++ b/ait/core/server/client.py
@@ -351,7 +351,7 @@ def __init__(
         else:
             raise (ValueError("TCPInputClient: Invalid Specification"))
 
-    def __exit__(self):
+    def __exit__(self, exc_type, exc_val, exc_tb):
         try:
             if self.sub:
                 self.sub.close()