diff --git a/scripts/upgrade b/scripts/upgrade
index 5f058f4..e1bf407 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -165,34 +165,9 @@ if [ "$upgrade_type" == "UPGRADE_APP" ]
 then
 	ynh_script_progression --message="Upgrading source files..."
 
-	# Create a temporary directory
-	tmpdir="$(mktemp -d)"
-
-	# Backup the config file in the temp dir
-	cp -af "$final_path/config/production.yaml" "$tmpdir/production.yaml"
-	if [ -s "$final_path/config/local-production.json" ]
-	then
-		cp -af "$final_path/config/local-production.json" "$tmpdir/local-production.json"
-	fi
-
-	# 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 the admin saved settings from tmp directory to final path
-	cp -af "$tmpdir/production.yaml" "$final_path/config/production.yaml"
-
-	if [ -s "$tmpdir/local-production.json" ]
-	then
-		cp -af "$tmpdir/local-production.json" "$final_path/config/local-production.json"
-	else
-		cp ../conf/local-production.json "$final_path/config/local-production.json"
-	fi
-
-	# Remove the tmp directory securely
-	ynh_secure_remove --file="$tmpdir"
+	ynh_setup_source --full_replace=1 --dest_dir="$final_path" \
+		--keep="config/production.yaml config/local-production.json config/local.yaml"
 fi
 
 chmod 750 "$final_path"