diff --git a/__init__.py b/__init__.py
index d0eaf58..67fb03a 100644
--- a/__init__.py
+++ b/__init__.py
@@ -369,7 +369,7 @@ def override_class_with_distorch(cls):
 NODE_CLASS_MAPPINGS = {"DeviceSelectorMultiGPU": DeviceSelectorMultiGPU}
 
 def check_module_exists(module_path):
-    full_path = os.path.join("custom_nodes", module_path)
+    full_path = os.path.join(folder_paths.get_folder_paths("custom_nodes")[0], module_path)
     logging.info(f"MultiGPU: Checking for module at {full_path}")
 
     if not os.path.exists(full_path):
@@ -1048,4 +1048,4 @@ if check_module_exists("ComfyUI-HunyuanVideoWrapper") or check_module_exists("co
     register_DownloadAndLoadHyVideoTextEncoder()
 
 
-logging.info(f"MultiGPU: Registration complete. Final mappings: {', '.join(NODE_CLASS_MAPPINGS.keys())}")
\ No newline at end of file
+logging.info(f"MultiGPU: Registration complete. Final mappings: {', '.join(NODE_CLASS_MAPPINGS.keys())}")