a few fixes

3 files changed, 9 insertions, 5 deletions

diff --git a/openbb_platform/providers/intrinio/openbb_intrinio/__init__.py b/openbb_platform/providers/intrinio/openbb_intrinio/__init__.py
index ce6e9515682..bf3e2701dbc 100644
--- a/openbb_platform/providers/intrinio/openbb_intrinio/__init__.py
+++ b/openbb_platform/providers/intrinio/openbb_intrinio/__init__.py
@@ -43,9 +43,7 @@ from openbb_intrinio.models.search_attributes import (
 )
 from openbb_intrinio.models.share_statistics import IntrinioShareStatisticsFetcher
 from openbb_intrinio.models.world_news import IntrinioWorldNewsFetcher
-from openbb_intrinio.models.zacks_sales_estimates import (
-    IntrinioZackSalesEstimatesFetcher,
-)
+from openbb_intrinio.models.sales_estimates import IntrinioZackSalesEstimatesFetcher
 
 intrinio_provider = Provider(
     name="intrinio",
diff --git a/openbb_platform/providers/intrinio/openbb_intrinio/models/zacks_sales_estimates.py b/openbb_platform/providers/intrinio/openbb_intrinio/models/sales_estimates.py
index 2c8188fd4ac..24c953ff414 100644
--- a/openbb_platform/providers/intrinio/openbb_intrinio/models/zacks_sales_estimates.py
+++ b/openbb_platform/providers/intrinio/openbb_intrinio/models/sales_estimates.py
@@ -8,6 +8,9 @@ from openbb_core.provider.standard_models.sales_estimates import (
     SalesEstimatesQueryParams,
 )
 from openbb_intrinio.utils.helpers import get_data_one
+from openbb_intrinio.utils.references import IntrinioCompany
+from pydantic import Field
+
 
 class IntrinioZackSalesEstimatesQueryParams(SalesEstimatesQueryParams):
     """Intrinio Zack Sales Estimates Query.
@@ -18,6 +21,9 @@ class IntrinioZackSalesEstimatesQueryParams(SalesEstimatesQueryParams):
 
 class IntrinioZackSalesEstimatesData(SalesEstimatesData):
     """Intrinio Zack Sales Estimates Data."""
+    company: Optional[IntrinioCompany] = Field(
+        description="The company to get sales estimates data from.", default=None
+    )
 
 
 class IntrinioZackSalesEstimatesFetcher(
@@ -34,7 +40,7 @@ class IntrinioZackSalesEstimatesFetcher(
         return IntrinioZackSalesEstimatesQueryParams(**params)
 
     @staticmethod
-     async def aextract_data(
+    async def aextract_data(
         query: IntrinioZackSalesEstimatesQueryParams,
         credentials: Optional[Dict[str, str]],
         **kwargs: Any,
diff --git a/openbb_platform/providers/intrinio/tests/test_intrinio_fetchers.py b/openbb_platform/providers/intrinio/tests/test_intrinio_fetchers.py
index 33c2bc6ea60..a82a2809a5c 100644
--- a/openbb_platform/providers/intrinio/tests/test_intrinio_fetchers.py
+++ b/openbb_platform/providers/intrinio/tests/test_intrinio_fetchers.py
@@ -46,7 +46,7 @@ from openbb_intrinio.models.search_attributes import (
 )
 from openbb_intrinio.models.share_statistics import IntrinioShareStatisticsFetcher
 from openbb_intrinio.models.world_news import IntrinioWorldNewsFetcher
-from openbb_intrinio.models.zacks_sales_estimates import (
+from openbb_intrinio.models.sales_estimates import (
     IntrinioZackSalesEstimatesFetcher,
 )