diff --git a/scripts/restore b/scripts/restore
index 557b788..ac89823 100755
--- a/scripts/restore
+++ b/scripts/restore
@@ -7,7 +7,7 @@ app=$YNH_APP_INSTANCE_NAME
 
 # The first argument is the backup directory location of the app
 # from where the script is executed
-backup_dir=$1
+backup_dir=$YNH_APP_BACKUP_DIR
 
 # Restore sources & data
 final_path="/var/www/${app}"