diff --git a/energon/nn/__init__.py b/energon/nn/__init__.py
index 74bb1f7..e8b23be 100644
--- a/energon/nn/__init__.py
+++ b/energon/nn/__init__.py
@@ -1,2 +1 @@
-from .layer import *
-from .wrapper import *
+from .layer import *
\ No newline at end of file