diff --git a/tests-unit/comfy_extras_test/__init__.py b/tests/unit/comfy_api_nodes_test/__init__.py
similarity index 100%
rename from tests-unit/comfy_extras_test/__init__.py
rename to tests/unit/comfy_api_nodes_test/__init__.py
diff --git a/tests-unit/comfy_api_nodes_test/mapper_utils_test.py b/tests/unit/comfy_api_nodes_test/mapper_utils_test.py
similarity index 100%
rename from tests-unit/comfy_api_nodes_test/mapper_utils_test.py
rename to tests/unit/comfy_api_nodes_test/mapper_utils_test.py
diff --git a/tests/unit/comfy_api_test/__init__.py b/tests/unit/comfy_api_test/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/tests-unit/comfy_api_test/input_impl_test.py b/tests/unit/comfy_api_test/input_impl_test.py
similarity index 100%
rename from tests-unit/comfy_api_test/input_impl_test.py
rename to tests/unit/comfy_api_test/input_impl_test.py
diff --git a/tests-unit/comfy_api_test/video_types_test.py b/tests/unit/comfy_api_test/video_types_test.py
similarity index 100%
rename from tests-unit/comfy_api_test/video_types_test.py
rename to tests/unit/comfy_api_test/video_types_test.py
diff --git a/tests/unit/comfy_extras_test/__init__.py b/tests/unit/comfy_extras_test/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/tests-unit/comfy_extras_test/image_stitch_test.py b/tests/unit/comfy_extras_test/image_stitch_test.py
similarity index 100%
rename from tests-unit/comfy_extras_test/image_stitch_test.py
rename to tests/unit/comfy_extras_test/image_stitch_test.py