diff --git a/e3nn_jax/_src/s2grid.py b/e3nn_jax/_src/s2grid.py
index eaad996..82cf137 100644
--- a/e3nn_jax/_src/s2grid.py
+++ b/e3nn_jax/_src/s2grid.py
@@ -1,5 +1,4 @@
 import math
-import sys
 from typing import Callable, List, Optional, Tuple, Union
 
 import jax