diff --git a/py/rvspecfit/desi/desi_fit.py b/py/rvspecfit/desi/desi_fit.py
index 71c138f..3dbd5d8 100644
--- a/py/rvspecfit/desi/desi_fit.py
+++ b/py/rvspecfit/desi/desi_fit.py
@@ -1359,7 +1359,7 @@ def submit(self, f, *args, **kw):
     def __enter__(self):
         return self
 
-    def __exit__(self):
+    def __exit__(self, exc_type, exc_value, traceback):
         pass