diff --git a/tests/_src/so3grid_test.py b/tests/_src/so3grid_test.py
index f100c17..13a1dc2 100644
--- a/tests/_src/so3grid_test.py
+++ b/tests/_src/so3grid_test.py
@@ -1,8 +1,6 @@
-import math
-
-import jax
 import jax.numpy as jnp
 import pytest
+
 from e3nn_jax import SO3Signal