diff --git a/waybar/.config/waybar/reboot_indicator.sh b/waybar/.config/waybar/reboot_indicator.sh
index f7c8cda..2fef006 100755
--- a/waybar/.config/waybar/reboot_indicator.sh
+++ b/waybar/.config/waybar/reboot_indicator.sh
@@ -1,8 +1,8 @@
 #!/bin/bash
 
 
-RUNNING_VERSION="$(uname -r | grep -o '[0-9]*\.[0-9]*\.[0-9]*-[0-9]*')"
-INSTALLED_VERSION="$(find /boot -maxdepth 0 -name 'vmlinuz*' | grep -o '[0-9]*\.[0-9]*\.[0-9]*-[0-9]*' | sort -Vr | head -1)"
+RUNNING_VERSION="$(uname -r)"
+INSTALLED_VERSION="$(find /boot -maxdepth 1 -name 'vmlinuz*' | sed 's/\/boot\///' | sort -Vr | head -1 | sed 's/vmlinuz-//')"
 
 #echo $RUNNING_VERSION
 #echo $INSTALLED_VERSION