diff --git a/setup-wireguard.sh b/setup-wireguard.sh
index b29b228..72c024f 100755
--- a/setup-wireguard.sh
+++ b/setup-wireguard.sh
@@ -18,6 +18,8 @@ sudo apt update
 
 if grep -Pq '^arm' <<< $(uname -m); then
     sudo apt install -y wireguard wireguard-dkms wireguard-tools raspberrypi-kernel raspberrypi-kernel-headers resolvconf
+elif [ -f "/run/.containerenv" ]; then
+    sudo apt install -y wireguard wireguard-tools resolvconf
 else
     sudo apt install -y wireguard wireguard-tools linux-headers-$(uname -r) resolvconf
 fi