diff --git a/data/unified_datasets/dailydialog/preprocess.py b/data/unified_datasets/dailydialog/preprocess.py
index e04a158782fa0e1df8d87a93838656c3ebb17a08..8a04aae5d620c813ed9be7c72b55d6d2c718fcdb 100644
--- a/data/unified_datasets/dailydialog/preprocess.py
+++ b/data/unified_datasets/dailydialog/preprocess.py
@@ -147,12 +147,10 @@ def preprocess():
     with ZipFile('data.zip', 'w', ZIP_DEFLATED) as zf:
         for filename in os.listdir(new_data_dir):
             zf.write(f'{new_data_dir}/{filename}')
-    # rmtree(original_data_dir)
-    # rmtree(new_data_dir)
+    rmtree(original_data_dir)
+    rmtree(new_data_dir)
     return dialogues, ontology
 
-    
-
 
 if __name__ == '__main__':
-    preprocess()
\ No newline at end of file
+    preprocess()