diff --git a/src/neris_api_client/client.py b/src/neris_api_client/client.py
index 0c4b63d..0e6ac56 100644
--- a/src/neris_api_client/client.py
+++ b/src/neris_api_client/client.py
@@ -387,7 +387,7 @@ def create_unit(
     ) -> Dict[str, Any]:
         return self._call(
             "post",
-            f"/entity/{neris_id_entity}/station/{neris_id_station}",
+            f"/entity/{neris_id_entity}/station/{neris_id_station}/unit",
             data=body,
             model=CreateUnitPayload,
         )
@@ -401,7 +401,7 @@ def patch_unit(
     ) -> Dict[str, Any]:
         return self._call(
             "patch",
-            f"/entity/{neris_id}/station/{neris_id_station}/unit/{neris_id_unit}",
+            f"/entity/{neris_id_entity}/station/{neris_id_station}/unit/{neris_id_unit}",
             data=body,
             model=PatchUnitPayload,
         )
diff --git a/tests/test_unit_routes.py b/tests/test_unit_routes.py
new file mode 100644
index 0000000..5f25211
--- /dev/null
+++ b/tests/test_unit_routes.py
@@ -0,0 +1,36 @@
+import unittest
+
+from neris_api_client.client import NerisApiClient
+
+
+class UnitRouteTests(unittest.TestCase):
+    def setUp(self):
+        self.client = object.__new__(NerisApiClient)
+        self.calls = []
+        self.client._call = lambda *args, **kwargs: self.calls.append((args, kwargs))
+
+    def test_create_unit_uses_unit_collection_route(self):
+        NerisApiClient.create_unit(self.client, "FD24000001", "ST24000001", {"name": "E1"})
+
+        self.assertEqual(
+            self.calls[0][0],
+            ("post", "/entity/FD24000001/station/ST24000001/unit"),
+        )
+
+    def test_patch_unit_uses_entity_id_parameter(self):
+        NerisApiClient.patch_unit(
+            self.client,
+            "FD24000001",
+            "ST24000001",
+            "UN24000001",
+            {"name": "E2"},
+        )
+
+        self.assertEqual(
+            self.calls[0][0],
+            ("patch", "/entity/FD24000001/station/ST24000001/unit/UN24000001"),
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()