diff --git a/scripts/install b/scripts/install
index b3910e6..9fc1f5a 100644
--- a/scripts/install
+++ b/scripts/install
@@ -8,7 +8,7 @@ source /usr/share/yunohost/helpers
 # Retrieve arguments
 domain=$YNH_APP_ARG_DOMAIN
 path=$YNH_APP_ARG_PATH
-is_public=$YNH_APP_ARG_PUBLIC_SITE
+is_public=$YNH_APP_ARG_IS_PUBLIC
 port=9001
 
 # Check domain/path availability
@@ -91,7 +91,7 @@ fi
 sudo service nginx reload
 # If app is public, add url to SSOWat conf as skipped_uris
 if [[ $is_public -eq 1 ]]; then
-  # unprotected_uris allows SSO credentials to be passed anyway.
+	# unprotected_uris allows SSO credentials to be passed anyway.
   ynh_app_setting_set "$app" unprotected_uris "/"
 fi
 
diff --git a/scripts/upgrade b/scripts/upgrade
index 6155ff3..108a737 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -10,6 +10,7 @@ domain=$(ynh_app_setting_get "$app" domain)
 path=$(ynh_app_setting_get "$app" path)
 dbuser=$(ynh_app_setting_get "$app" dbuser)
 dbname=$(ynh_app_setting_get "$app" dbname)
+is_public=$(ynh_app_setting_get "$app" is_public)
 
 root_pwd=$(sudo cat /etc/yunohost/mysql)