diff --git a/scripts/install b/scripts/install
index aac1dbc..d3c1e17 100755
--- a/scripts/install
+++ b/scripts/install
@@ -140,15 +140,12 @@ if [ $generate_custom_world -eq 1 ]; then
 	ynh_script_progression --message="Generating custom world..." --weight=100
 
 	generate_custom_world
+else
+	ynh_script_progression --message="Adding a configuration file..."
+
+	add_configuration_files
 fi
 
-#=================================================
-# ADD A CONFIGURATION
-#=================================================
-ynh_script_progression --message="Adding a configuration file..."
-
-add_configuration_files
-
 #=================================================
 # SETUP SYSTEMD
 #=================================================
diff --git a/scripts/upgrade b/scripts/upgrade
index 1af93c7..451f522 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -99,15 +99,12 @@ if [ $generate_custom_world -eq 1 ]; then
 	ynh_script_progression --message="Generating custom world..." --weight=100
 
 	generate_custom_world
+else
+	ynh_script_progression --message="Adding a configuration file..."
+
+	add_configuration_files
 fi
 
-#=================================================
-# UPDATE A CONFIG FILE
-#=================================================
-ynh_script_progression --message="Updating a configuration file..."
-
-add_configuration_files
-
 #=================================================
 # SETUP SYSTEMD
 #=================================================