diff --git a/src/elisa/__init__.py b/src/elisa/__init__.py
index a7ee4046..4899f604 100644
--- a/src/elisa/__init__.py
+++ b/src/elisa/__init__.py
@@ -1,8 +1,5 @@
-from .data import *
-from .model import *
-from .infer import *
-from .util import *
-from .__about__ import __version__
+from .model import *  # noqa: F403
+from .util import jax_enable_x64, set_cpu_cores
 
 jax_enable_x64(True)
 set_cpu_cores(4)
diff --git a/src/elisa/util/__init__.py b/src/elisa/util/__init__.py
index e69de29b..907ef6c8 100644
--- a/src/elisa/util/__init__.py
+++ b/src/elisa/util/__init__.py
@@ -0,0 +1 @@
+from .config import *  # noqa: F403