diff --git a/imperative/python/megengine/traced_module/__init__.py b/imperative/python/megengine/traced_module/__init__.py
index 4256f5e033bef0b64699b376d381a005ed43ca63..c906b87905b0a02d12d05b62531234484054aec9 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 7bc157057bb9bd95da614f34d252186b9a80f4f3..c16ecdb024c8bdcc958df792339c9a1b8d3483d2 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,