diff --git a/tests/fused_attention_stablehlo_test.py b/tests/fused_attention_stablehlo_test.py
index 83e2fd350..af0b18b02 100644
--- a/tests/fused_attention_stablehlo_test.py
+++ b/tests/fused_attention_stablehlo_test.py
@@ -14,7 +14,6 @@
 
 from functools import partial
 from absl.testing import absltest
-import os
 
 import numpy as np
 import jax