diff --git a/src/evox/problems/neuroevolution/supervised_learning/tfds.py b/src/evox/problems/neuroevolution/supervised_learning/tfds.py
index b8fc7754..c9e36eca 100644
--- a/src/evox/problems/neuroevolution/supervised_learning/tfds.py
+++ b/src/evox/problems/neuroevolution/supervised_learning/tfds.py
@@ -8,7 +8,7 @@
 import tensorflow_datasets as tfds
 from jax.tree_util import tree_map
 
-from evox import Problem, Static, dataclass, jit_class
+from evox import Problem, dataclass, jit_class
 from evox.utils.io import x32_func_call