diff --git a/setup_data_and_models.sh b/setup_data_and_models.sh
index bd4a708..4170713 100644
--- a/setup_data_and_models.sh
+++ b/setup_data_and_models.sh
@@ -12,4 +12,4 @@ mv out/pretrained/* out/
 rm data/dataset.zip 
 rm out/pretrained.zip
 rm -r data/dataset 
-rm -r data/pretrained
+rm -r out/pretrained