diff --git a/scripts/remove b/scripts/remove
index 08c441a..e8575b9 100644
--- a/scripts/remove
+++ b/scripts/remove
@@ -111,7 +111,7 @@ ynh_script_progression --message="Removing dependencies..."
 # Remove metapackage and its dependencies
 ynh_remove_app_dependencies
 
-# Remove the line for this app
+# Remove Prosody
 ynh_app_setting_delete --app=$app --key=require_prosody
 
 # List apps requiring Prosody
diff --git a/scripts/upgrade b/scripts/upgrade
index 15b9b2b..bdeac96 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -134,6 +134,8 @@ else
 	yunohost app upgrade prosody --force
 fi
 
+ynh_app_setting_set --app=$app --key=require_prosody --value="1"
+
 #=================================================
 # SPECIFIC UPGRADE
 #=================================================