diff --git a/tests/compile/piecewise/test_toy_llama.py b/tests/compile/piecewise/test_toy_llama.py
index 075877bb6652e..f95932c498e9b 100644
--- a/tests/compile/piecewise/test_toy_llama.py
+++ b/tests/compile/piecewise/test_toy_llama.py
@@ -470,4 +470,4 @@ def benchmark():
 
 
 if __name__ == "__main__":
-    pass
\ No newline at end of file
+    benchmark()