diff --git a/vector/app/main.py b/vector/app/main.py
index ecaab0c..2a7910d 100644
--- a/vector/app/main.py
+++ b/vector/app/main.py
@@ -43,7 +43,8 @@ from app.models.vector_models import (
     ActionRecommendationRequest, ActionRecommendationResponse,
     ActionRecommendationSimpleResponse,
     VectorDBStatusResponse, VectorDBStatus,
-    FindReviewsResponse, StoredDataInfo
+    FindReviewsResponse, StoredDataInfo, 
+    StoreInfoRequest, StoreInfoResponse
 )
 
 from app.services.restaurant_service import RestaurantService