diff --git a/scripts/upgrade b/scripts/upgrade
index a2fa9cc..e9e9f02 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -124,7 +124,7 @@ then
 	ynh_script_progression --message="Upgrading source files..." --weight=3
 
 	# Download, check integrity, uncompress and patch the source from app.src
-	ynh_setup_source --dest_dir="$final_path" --keep="Specific/ config/"
+	ynh_setup_source --dest_dir="$final_path" --keep="config/baikal.yaml"
 fi
 
 chmod 750 "$final_path"
@@ -162,16 +162,16 @@ ynh_add_nginx_config
 
 # We keep this to allow upgrade the config file in case it needs to be changed.
 
-#if [ "$upgrade_type" == "UPGRADE_APP" ]
-#then
-#	ynh_script_progression --message="Upgrading the configuration file..." --weight=2
+if [ "$upgrade_type" == "UPGRADE_APP" ]
+then
+	ynh_script_progression --message="Upgrading the configuration file..." --weight=2
 
-#	timezone=$(cat /etc/timezone)
-#	path=${path_url%/}
-#	ynh_add_config --template="../conf/baikal.yaml" --destination="$final_path/config/baikal.yaml"
-#	chown $app: "$final_path/config/baikal.yaml"
-#	chmod 640 "$final_path/config/baikal.yaml"
-#fi
+	timezone=$(cat /etc/timezone)
+	path=${path_url%/}
+	ynh_add_config --template="../conf/baikal.yaml" --destination="$final_path/config/baikal.yaml"
+	chown $app: "$final_path/config/baikal.yaml"
+	chmod 640 "$final_path/config/baikal.yaml"
+fi
 
 #=================================================
 # GENERIC FINALIZATION