diff --git a/treex/nn/__init__.py b/treex/nn/__init__.py
index 41218aaf..da9f416b 100644
--- a/treex/nn/__init__.py
+++ b/treex/nn/__init__.py
@@ -5,7 +5,6 @@
 from .dropout import Dropout
 from .flatten import Flatten
 from .flax_module import FlaxModule
-from .haiku_module import HaikuModule
 from .linear import Linear
 from .mlp import MLP
 from .sequential import Lambda, Sequential, sequence