diff --git a/unzip_data.py b/unzip_data.py
index e39fd3e9aa2de85a63ce111bd67ab75556dcced3..09ac74c978edaab33df1fea7f16d1776f492e9b5 100644
--- a/unzip_data.py
+++ b/unzip_data.py
@@ -13,11 +13,11 @@ def unzip_and_remove(zip_filename):
             zip_filename,
         )
     files = zf.ZipFile(zip_filename, 'r')
-    files.extractall('.')
+    files.extractall('mlsmm2156')
     files.close()
     os.remove(zip_filename)
-    shutil.rmtree('__MACOSX')  # remove macosx data for Mac users
+    shutil.rmtree('mlsmm2156/__MACOSX')  # remove macosx data for Mac users
 
 
 if __name__ == "__main__":
-    unzip_and_remove("data.zip")
\ No newline at end of file
+    unzip_and_remove("mlsmm2156/data.zip")