diff --git a/scripts/i3blocks/vpn b/scripts/i3blocks/vpn
index dbaed98..8bb6a93 100755
--- a/scripts/i3blocks/vpn
+++ b/scripts/i3blocks/vpn
@@ -3,12 +3,10 @@
 INSTANCE="${BLOCK_INSTANCE:-}"
 
 # Check VPN status
-GET_VPN=$(ip route | grep -E -- "tun0|$INSTANCE" | awk '{ print $3 }')
+GET_VPN=$(ip route | grep -E -- "tun0|wg0|$INSTANCE" | awk '{ print $3 }')
 
 # Store status
-if [[ $GET_VPN == *"tun0"* ||  $GET_VPN == "$INSTANCE" ]]
-then
-    echo -e "<span color='#00FF00'>ON</span>\n"
-else
-    echo -e "<span color='#FFDD00'>OFF</span>\n"
-fi
+case $GET_VPN in
+    tun0|wg0|$INSTANCE) echo -e "<span color='#00FF00'>ON</span>\n";;
+    *) echo -e "<span color='#FFDD00'>OFF</span>\n";;
+esac