diff --git a/modules/cli/default.nix b/modules/cli/default.nix
index 7085f27..651507f 100644
--- a/modules/cli/default.nix
+++ b/modules/cli/default.nix
@@ -48,8 +48,8 @@
     (writeShellScriptBin "nixos-check-reboot" ''
       set -e
       booted="$(readlink /run/booted-system/kernel)"
-      built="$(readlink /nix/var/nix/profiles/system/kernel)"
-      if [ "$booted" = "$built" ]
+      current="$(readlink /run/current-system/kernel)"
+      if [ "$booted" = "$current" ]
       then
           echo OK
           exit 0