diff --git a/scripts/upgrade b/scripts/upgrade
index 17eabe9..e67fbb7 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -50,6 +50,22 @@ fi
 #=================================================
 ynh_script_progression --message="Ensuring downward compatibility..."
 
+#=================================================
+# SPECIFIC SETUP
+#=================================================
+# CONFIGURE MONGOD
+#=================================================
+ynh_script_progression --message="Configuring MongoDB..." --weight=10
+
+ynh_replace_string --match_string="#replication:" --replace_string="replication:\n  replSetName: rs01" --target_file="/etc/mongod.conf"
+
+ynh_exec_warn_less systemctl enable mongod --quiet
+ynh_systemd_action --service_name=mongod --action=restart --log_path=/var/log/mongodb/mongod.log --line_match="Waiting for connections"
+
+if ynh_exec_warn_less ynh_mongo_exec --command="printjson(rs.status())" | grep -q "no replset config has been received"; then
+	ynh_exec_warn_less ynh_mongo_exec --command="printjson(rs.initiate())" --eval
+fi
+
 #=================================================
 # DOWNLOAD, CHECK AND UNPACK SOURCE
 #=================================================