diff --git a/scripts/upgrade b/scripts/upgrade
index e3fe4d8..5818d7a 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -38,7 +38,7 @@ then
 	ynh_script_progression --message="Upgrading source files..." --weight=5
 
 	# Download, check integrity, uncompress and patch the source from app.src
-	ynh_setup_source --dest_dir="$install_dir" --keep="config.local.user.php data" --full_replace=1
+	ynh_setup_source --dest_dir="$install_dir" --keep="association.sqlite config.local.user.php data"
 fi
 
 chmod -R o-rwx "$install_dir"
@@ -98,15 +98,13 @@ chmod 660 "$install_dir/config.local.user.php"
 #=================================================
 ynh_script_progression --message="Finalise upgrade" --weight=1
 
-# Move the database on folder data/data created and remove the folder
-mv $install_dir/data/data/association.sqlite $install_dir/data/
-ynh_secure_remove --file=$install_dir/data/data
-
 ynh_local_curl "/admin/index.php"
 sleep 5
 ynh_local_curl "/index.php"
 sleep 5
 
+# Move the database on folder data/data created and remove the folder
+ynh_secure_remove --file=$install_dir/data/data
 #=================================================
 # END OF SCRIPT
 #=================================================