diff --git a/scripts/install b/scripts/install
index d7a0311..714857b 100755
--- a/scripts/install
+++ b/scripts/install
@@ -46,6 +46,7 @@ ynh_script_progression --message="Storing installation settings..." --weight=2
 
 ynh_app_setting_set --app=$app --key=domain --value=$domain
 ynh_app_setting_set --app=$app --key=path --value=$path_url
+ynh_app_setting_set --app=$app --key=maindomain --value=$maindomain
 
 #=================================================
 # CREATE DEDICATED USER
diff --git a/scripts/upgrade b/scripts/upgrade
index 7344212..00750d2 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -18,6 +18,7 @@ app=$YNH_APP_INSTANCE_NAME
 domain=$(ynh_app_setting_get --app=$app --key=domain)
 path_url=$(ynh_app_setting_get --app=$app --key=path)
 final_path=$(ynh_app_setting_get --app=$app --key=final_path)
+maindomain=$(ynh_app_setting_get --app=$app --key=maindomain)
 
 #=================================================
 # CHECK VERSION
@@ -50,6 +51,12 @@ if [ -z "$final_path" ]; then
 	ynh_app_setting_set --app=$app --key=final_path --value=$final_path
 fi
 
+# If maindomain doesn't exist, create it
+if [ -z "$maindomain" ]; then
+	maindomain=$(yunohost domain main-domain --output-as plain)
+	ynh_app_setting_set --app=$app --key=maindomain --value=$maindomain
+fi
+
 # Cleaning legacy permissions
 if ynh_legacy_permissions_exists; then
 	ynh_legacy_permissions_delete_all