diff --git a/sharktank/tests/evaluate/perplexity_test.py b/sharktank/tests/evaluate/perplexity_test.py
index 3351f1aaf..e035e55eb 100644
--- a/sharktank/tests/evaluate/perplexity_test.py
+++ b/sharktank/tests/evaluate/perplexity_test.py
@@ -15,7 +15,7 @@
 logger.setLevel(logging.INFO)
 
 
-class PerplexityTest(unittest.TestCase):
+class PerplexityTest:
     @pytest.mark.expensive
     @pytest.mark.integration
     def test(self):
@@ -49,4 +49,4 @@ def test(self):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    PerplexityTest().test()