diff --git a/templates/template.cpp b/templates/template.cpp
index ee26ee6..c65ef2c 100644
--- a/templates/template.cpp
+++ b/templates/template.cpp
@@ -96,13 +96,7 @@ int main()
     pre();
 
     int t;
-
-#if TEST == 1
-    cin >> t;
-#else
-    t = 1;
-#endif
-
+    (TEST && cin >> t) || (t = 1);
     while (t--) {
         solve();
     }