diff --git a/sjrpc/utils/datastructures.py b/sjrpc/utils/datastructures.py
index 0715d171f8983c3c03109220fa797e3e8027070e..31d6fee497145cb7d5d8a2576ed2317c2ec890cc 100644
--- a/sjrpc/utils/datastructures.py
+++ b/sjrpc/utils/datastructures.py
@@ -67,5 +67,5 @@ class BytesBuffer(object):
     def __enter__(self):
         return self._lock.__enter__()
 
-    def __exit__(self):
-        return self._lock.__exit__()
+    def __exit__(self, exc_type, exc_value, traceback):
+        return self._lock.__exit__(exc_type, exc_value, traceback)