diff --git a/scripts/restore b/scripts/restore
index d58b39b..3d3aef5 100644
--- a/scripts/restore
+++ b/scripts/restore
@@ -18,7 +18,7 @@
 
 # Check domain/path availability
 	sudo yunohost app checkurl "${domain}${path}" -a "$app" \
-	  || exit 1
+	  || ynh_die
 
 # Check destination directory
 	DESTDIR="/var/www/$app"