diff --git a/plai/core/module.py b/plai/core/module.py
index 67d1e29..5c7cb9f 100644
--- a/plai/core/module.py
+++ b/plai/core/module.py
@@ -38,7 +38,7 @@ def __init_subclass__(cls, **kwargs):
         Node.subclass_dict[op_name] = cls
 
     @staticmethod
-    def get_op_subclass(op_name: str):
+    def get_node_class(op_name: str):
         assert op_name in Node.subclass_dict, f'Unregister Class with name: {op_name}'
         return Node.subclass_dict[op_name]