diff --git a/test/srt/test_model_hooks.py b/test/srt/test_model_hooks.py
index f020f74da..656b124d0 100644
--- a/test/srt/test_model_hooks.py
+++ b/test/srt/test_model_hooks.py
@@ -1,6 +1,5 @@
 import argparse
 import json
-import unittest
 
 import torch
 import torch.nn as nn
@@ -149,4 +148,5 @@ class TestAttachHooks(CustomTestCase):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    pass
+    # unittest.main()