diff --git a/scripts/upgrade b/scripts/upgrade
index 1acf661..1b20b74 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -10,6 +10,7 @@ source _common.sh
 source ynh_package_version
 source ynh_detect_arch__2
 source ynh_add_config
+source ynh_mongo_db
 source /usr/share/yunohost/helpers
 
 #=================================================