diff --git a/tests/test_flash_mla_prefill.py b/tests/test_flash_mla_prefill.py
index d2f5b7e6..5c92562e 100644
--- a/tests/test_flash_mla_prefill.py
+++ b/tests/test_flash_mla_prefill.py
@@ -1,5 +1,4 @@
 import math
-import time
 from typing import Tuple
 import random
 import dataclasses