diff --git a/src/creative_agent/server.py b/src/creative_agent/server.py
index 70d3924..823587b 100644
--- a/src/creative_agent/server.py
+++ b/src/creative_agent/server.py
@@ -74,12 +74,12 @@ def list_creative_formats(
         )
 
         # Import required types
-        from .schemas_generated._schemas_v1_media_buy_list_creative_formats_response_json import (
+        from .schemas_generated._schemas_v1_creative_list_creative_formats_response_json import (
             Capability,
             CreativeAgent,
             Status,
         )
-        from .schemas_generated._schemas_v1_media_buy_list_creative_formats_response_json import (
+        from .schemas_generated._schemas_v1_creative_list_creative_formats_response_json import (
             Format as ResponseFormat,
         )