diff --git a/scripts/upgrade b/scripts/upgrade
index b1cf9d4..647b197 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -10,6 +10,7 @@ source _common.sh
 source ynh_add_extra_apt_repos__3
 source ynh_install_php__3
 source ynh_composer__2
+source ynh_add_swap
 source /usr/share/yunohost/helpers
 
 #=================================================