Blob Blame Raw
diff --git a/q.py b/q.py
index 1bf76d0..fcff4c9 100644
--- a/q.py
+++ b/q.py
@@ -294,8 +294,9 @@ def __call__(self, *args):
             lines = info.code_context[:info.index + 1]
 
         # If we see "@q" on a single line, behave like a trace decorator.
-        if lines[-1].strip().startswith('@') and args:
-            return self.trace(args[0])
+        for line in lines:
+            if line.strip() in ('@q', '@q()') and args:
+                return self.trace(args[0])
 
         # Otherwise, search for the beginning of the call expression; once it
         # parses, use the expressions in the call to label the debugging output.