diff --git a/tests/python/tir-analysis/test_tir_analysis_calculate_workspace.py b/tests/python/tir-analysis/test_tir_analysis_calculate_workspace.py
index 00e9df9d8b34..29bfc5845870 100644
--- a/tests/python/tir-analysis/test_tir_analysis_calculate_workspace.py
+++ b/tests/python/tir-analysis/test_tir_analysis_calculate_workspace.py
@@ -123,5 +123,4 @@ def test_vector_type():
 
 
 if __name__ == "__main__":
-    test_global_allocates()
-    test_local_allocates()
+    tvm.testing.main()