diff --git a/autoload/kwin-get-class.nu b/autoload/kwin-get-class.nu
index da2a25f..2ae89f5 100644
--- a/autoload/kwin-get-class.nu
+++ b/autoload/kwin-get-class.nu
@@ -1,7 +1,11 @@
 export def lumen-kwin-get-class [waittimesec: int]: nothing -> string {
-  for $i in $waittimesec..1 {
-    $i | print
-    sleep 1sec
+  if ($waittimesec == 0) {
+    qdbus6 org.kde.KWin /KWin queryWindowInfo
+  } else {
+    for $i in $waittimesec..1 {
+      $i | print
+      sleep 1sec
+    }
+    qdbus6 org.kde.KWin /KWin queryWindowInfo
   }
-  qdbus6 org.kde.KWin /KWin queryWindowInfo
 }