diff --git a/scripts/upgrade b/scripts/upgrade
index 6cb6a86..d5c6305 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -24,7 +24,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="$install_dir" --keep="Specific config"
+	ynh_setup_source --dest_dir="$install_dir" --keep="config/baikal.yaml"
 fi
 
 chmod -R o-rwx "$install_dir"
@@ -49,16 +49,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%/}
-#	ynh_add_config --template="../conf/baikal.yaml" --destination="$install_dir/config/baikal.yaml"
-#	chown $app: "$install_dir/config/baikal.yaml"
-#	chmod 640 "$install_dir/config/baikal.yaml"
-#fi
+	timezone=$(cat /etc/timezone)
+	path=${path%/}
+	ynh_add_config --template="../conf/baikal.yaml" --destination="$install_dir/config/baikal.yaml"
+	chown $app: "$install_dir/config/baikal.yaml"
+	chmod 640 "$install_dir/config/baikal.yaml"
+fi
 
 #=================================================
 # END OF SCRIPT