diff --git a/api/endpoints/idea_router.py b/api/endpoints/idea_router.py
index a9b612d..c78c5d4 100644
--- a/api/endpoints/idea_router.py
+++ b/api/endpoints/idea_router.py
@@ -5,7 +5,8 @@ from api.endpoints.auth import get_current_user
 from api.service.idea_service import IdeaService
 from api.service.generation_service import GenerationService
 from models.Idea import Idea
-from api.models.GenerationRequest import GenerationResponse, GenerationsResponse
+from api.models import GenerationResponse, GenerationsResponse
+from api.models import IdeaRequest, PostRequest  # Adjusting for general model usage
 from api.models.IdeaRequest import IdeaCreateRequest, IdeaUpdateRequest, IdeaResponse
 
 router = APIRouter(prefix="/api/ideas", tags=["ideas"])