diff --git a/setup/restore_database.sh b/setup/restore_database.sh
index a570dc8bd..bb18ac0cb 100755
--- a/setup/restore_database.sh
+++ b/setup/restore_database.sh
@@ -47,14 +47,18 @@ ColorReset() {
 }
 
 
-main(){
+main() {
     DEPLOYMENT_DIR=$(pwd)/deployment
+    BACKUP_DIR="$DEPLOYMENT_DIR/backup"
+    BACKUP_FILE="$BACKUP_DIR/backup.zip"
     DOWNLOAD_URL="$1"
     CONTAINER_ID="$2"
-    BACKUP_FILE="$DEPLOYMENT_DIR/backup/backup.zip"
 
-    # Create the deployment directory if it doesn't exist
-    mkdir -p "$DEPLOYMENT_DIR/backup"
+    # Check if the deployment and backup directories exist, create them if not
+    if [[ ! -d "$BACKUP_DIR" ]]; then
+        echo "Backup directory does not exist. Creating: $BACKUP_DIR"
+        mkdir -p "$BACKUP_DIR"
+    fi
 
     # Check if the download URL is valid
     echo "Checking if the URL is valid: $DOWNLOAD_URL"
@@ -65,8 +69,8 @@ main(){
         exit 1
     fi
 
-    # Download the file
-    wget -P "$DEPLOYMENT_DIR/backup" -O backup.zip "$DOWNLOAD_URL"
+    # Download the file and rename it to backup.zip
+    wget -O "$BACKUP_FILE" "$DOWNLOAD_URL"
 
     # Check if the file was downloaded
     if [[ -f "$BACKUP_FILE" ]]; then
@@ -81,4 +85,4 @@ main(){
     sudo docker exec "${CONTAINER_ID}" odoo db --config=/etc/odoo/odoo.conf load backup /etc/odoo/backup/backup.zip
 }
 
-main $1 $2
\ No newline at end of file
+main "$@"
\ No newline at end of file