diff --git a/imperative/python/megengine/traced_module/__init__.py b/imperative/python/megengine/traced_module/__init__.py
index 4256f5e0..c906b879 100644
--- a/imperative/python/megengine/traced_module/__init__.py
+++ b/imperative/python/megengine/traced_module/__init__.py
@@ -21,10 +21,10 @@ from .traced_module import (
 _register_all_builtin_module()
 set_cpp_apply_module_trace(cpp_apply_module_trace)
 
-__all__ = {
+__all__ = [
     "register_as_builtin",
     "trace_module",
     "wrap",
     "TracedModule",
     "optimize",
-}
+]
diff --git a/imperative/python/megengine/traced_module/node.py b/imperative/python/megengine/traced_module/node.py
index 96091b36..4bfff4af 100644
--- a/imperative/python/megengine/traced_module/node.py
+++ b/imperative/python/megengine/traced_module/node.py
@@ -222,7 +222,7 @@ class TensorNode(Node):
 
     def __init__(
         self,
-        expr: "Expr",
+        expr,
         name: str = None,
         qualname: str = None,
         shape: Tuple[int] = None,