diff --git a/buggy.py b/buggy.py
--- a/buggy.py
+++ b/buggy.py
@@ -5,7 +5,7 @@
     t = int(input())
     for _ in range(t):
         k = int(input())
-        if k == 1:
+        if k == 1 or k == 2:
             print(1)
             continue
 
