diff --git a/mapper.py b/mapper.py
index 2c4f6b6..66a7657 100644
--- a/mapper.py
+++ b/mapper.py
@@ -20,7 +20,7 @@ class NodePackageMapper:
                 # "custom_nodes.PackageName" -> "PackageName"
                 # "comfy_extras.nodes_xyz" -> "__builtin__"
                 # "comfy_api_nodes.xyz" -> "__builtin__"
-                parts = module.split(".", 1)
+                parts = module.split(".")
                 if parts[0] == "custom_nodes" and len(parts) > 1:
                     self._map[class_type] = parts[1]
                 else: