diff --git a/integrations/astra/src/haystack_integrations/document_stores/astra/astra_client.py b/integrations/astra/src/haystack_integrations/document_stores/astra/astra_client.py
index 4272c69d1..2549719d6 100644
--- a/integrations/astra/src/haystack_integrations/document_stores/astra/astra_client.py
+++ b/integrations/astra/src/haystack_integrations/document_stores/astra/astra_client.py
@@ -5,12 +5,11 @@
 
 from astrapy.api import APIRequestError
 from astrapy.db import AstraDB
-from pydantic.dataclasses import dataclass
 
 # For version tracking
 from haystack import __name__ as integration_name
 from haystack.version import __version__ as integration_version
-
+from pydantic.dataclasses import dataclass
 
 logger = logging.getLogger(__name__)