diff --git a/google/cloud/pubsublite/internal/endpoints.py b/google/cloud/pubsublite/internal/endpoints.py
index 2837d18b..0422e1c7 100644
--- a/google/cloud/pubsublite/internal/endpoints.py
+++ b/google/cloud/pubsublite/internal/endpoints.py
@@ -1,4 +1,4 @@
-from google.cloud.pubsublite.location import CloudRegion
+from google.cloud.pubsublite.types import CloudRegion
 
 
 def regional_endpoint(region: CloudRegion):
diff --git a/google/cloud/pubsublite/internal/routing_metadata.py b/google/cloud/pubsublite/internal/routing_metadata.py
index ae8e6d00..d29aa882 100644
--- a/google/cloud/pubsublite/internal/routing_metadata.py
+++ b/google/cloud/pubsublite/internal/routing_metadata.py
@@ -1,8 +1,7 @@
 from typing import Mapping
 from urllib.parse import urlencode
 
-from google.cloud.pubsublite.partition import Partition
-from google.cloud.pubsublite.paths import TopicPath, SubscriptionPath
+from google.cloud.pubsublite.types import Partition, TopicPath, SubscriptionPath
 
 _PARAMS_HEADER = "x-goog-request-params"