diff --git a/model-weights/get_models.sh b/model-weights/get_models.sh
index 79a27ac..084b748 100644
--- a/model-weights/get_models.sh
+++ b/model-weights/get_models.sh
@@ -17,10 +17,7 @@ wget --load-cookies /tmp/cookies.txt -r "https://docs.google.com/uc?export=downl
 wget --load-cookies /tmp/cookies.txt -r "https://docs.google.com/uc?export=download&confirm=$(wget --quiet --save-cookies /tmp/cookies.txt --keep-session-cookies --no-check-certificate 'https://docs.google.com/uc?export=download&id=1a_pbXPYNj7_Gi6OxUqNo_T23Dt_9CzOV' -O- | sed -rn 's/.*confirm=([0-9A-Za-z_]+).*/\1\n/p')&id=1a_pbXPYNj7_Gi6OxUqNo_T23Dt_9CzOV" -O $H5_FILENAME && rm -rf /tmp/cookies.txt
 
 # Unzip
-unzip -q $W_FILENAME
-unzip -q $H5_FILENAME
-
-# Delete .zip files
-rm -rf $W_FILENAME $H5_FILENAME
+unzip -q $W_FILENAME && rm -rf $W_FILENAME
+unzip -q $H5_FILENAME && rm -rf $H5_FILENAME
 
 echo "*** All done!!!"