diff --git a/mlair/data_handler/default_data_handler.py b/mlair/data_handler/default_data_handler.py
index c7bfee79513c9cad67a1f9c68eb37e327d414296..51391bf35201c3f655f44a2d424a2134f3efa0e8 100644
--- a/mlair/data_handler/default_data_handler.py
+++ b/mlair/data_handler/default_data_handler.py
@@ -84,7 +84,7 @@ class DefaultDataHandler(AbstractDataHandler):
     def _cleanup(self):
         directory = os.path.dirname(self._save_file)
         if os.path.exists(directory) is False:
-            os.makedirs(directory)
+            os.makedirs(directory, exist_ok=True)
         if os.path.exists(self._save_file):
             shutil.rmtree(self._save_file, ignore_errors=True)