diff --git a/scripts/upgrade b/scripts/upgrade
index 3610b7a..bc2f4a6 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -109,9 +109,24 @@ ynh_system_user_create --username=$app --home_dir="$final_path"
 if [ "$upgrade_type" == "UPGRADE_APP" ]
 then
 	ynh_script_progression --message="Upgrading source files..." --time --weight=1
+	
+	# Create a temporary directory
+	tmpdir="$(mktemp -d)"
+	
+	# Backup the inc/digiwords.db file to the temp dir
+	cp -ar "$final_path/inc/digiwords.db" "$tmpdir/digiwords.db"
+	
+	# Remove the app directory securely
+	ynh_secure_remove --file=$final_path
 
 	# Download, check integrity, uncompress and patch the source from app.src
 	ynh_setup_source --dest_dir="$final_path"
+	
+	# Copy digiwords.db back to the final_path
+	cp -ar "$tmpdir/digiwords.db" "$final_path/inc/digiwords.db"
+
+	# Remove the tmp directory securely
+	ynh_secure_remove --file="$tmpdir"
 fi
 
 # FIXME: this should be managed by the core in the future
@@ -150,10 +165,6 @@ ynh_add_fpm_config
 #=================================================
 # SPECIFIC UPGRADE
 #=================================================
-# TAKE BACK THE OLD digiwords.db file
-#=================================================
-ynh_script_progression --message="Restoring database..." --time --weight=1
-ynh_restore_file -o "$final_path/inc/digiwords.db"
 
 #=================================================
 # GENERIC FINALIZATION