diff --git a/scripts/upgrade b/scripts/upgrade
index bbd0dc3..ef4e17d 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -41,12 +41,12 @@ ynh_systemd_action --service_name=$app --action="stop" --log_path="/var/log/$app
 # DOWNLOAD, CHECK AND UNPACK SOURCE
 #=================================================
 
-if [ $(ynh_compare_current_package_version --comparison le --version 2.6.2~ynh4) ]; then
+if [ $(ynh_compare_current_package_version --comparison le --version "2.6.2~ynh4") ]; then
 	ynh_script_progression --message="Removing old files..." --weight=2
 	rm -rf "$install_dir/{*,.*}"
 fi
 
-if [[ "$upgrade_type" == "UPGRADE_APP" || $(ynh_compare_current_package_version --comparison le --version 2.6.2~ynh4) ]]; then
+if [[ "$upgrade_type" == "UPGRADE_APP" || $(ynh_compare_current_package_version --comparison le --version "2.6.2~ynh4") ]]; then
 	ynh_script_progression --message="Upgrading source files..." --weight=5
 	# Download, check integrity, uncompress and patch the source from app.src
 	ynh_setup_source --dest_dir="$install_dir/build" --full_replace=1 #--keep=".env"