diff --git a/scripts/restore b/scripts/restore
index 6bb9078..b58a574 100644
--- a/scripts/restore
+++ b/scripts/restore
@@ -44,7 +44,7 @@ swap_needed=0
 
 if [ $total_memory -lt $memory_needed ]; then
 	# Need a minimum of memory
-	if [ -n "$(grep 'kthreadd' /proc/2/status 2>/dev/null)" ]; then
+	if [ "$(systemd-detect-virt)" != "lxc" ]; then
 		swap_needed=$(($memory_needed - $total_memory))
 		ynh_script_progression --message="Adding $swap_needed Mo to swap..."
 		ynh_add_swap --size=$swap_needed