diff --git a/byteps/mxnet/__init__.py b/byteps/mxnet/__init__.py
index e880f150e..9c14c9f92 100644
--- a/byteps/mxnet/__init__.py
+++ b/byteps/mxnet/__init__.py
@@ -230,9 +230,6 @@ def __init__(self, params, optimizer, optimizer_params=None, root_rank=0, compre
                 )
                 byteps_declare_tensor("gradient_" + str(i), **byteps_params)
 
-    def __del__(self):
-        self._f.close()
-
     def _register_compressor(self, params, optimizer_params, compression_params):
         """Register compressor for BytePS
 
diff --git a/byteps/mxnet/compression.py b/byteps/mxnet/compression.py
index fd90f2cb3..fab9a8ff6 100644
--- a/byteps/mxnet/compression.py
+++ b/byteps/mxnet/compression.py
@@ -105,8 +105,8 @@ def decompress(self, tensor, ctx, *args, **kwargs):
         try:
             self.future.result(timeout=0.1)
             tensor += self.cache
-        except TimeoutError:
-            print("timeout")
+        except concurrent.futures.TimeoutError:
+            print("timeout for wd-momentum")
         return self.compressor.decompress(tensor, ctx)