diff --git a/integrations/amazon_bedrock/src/amazon_bedrock_haystack/generators/amazon_bedrock.py b/integrations/amazon_bedrock/src/amazon_bedrock_haystack/generators/amazon_bedrock.py
index fc96b1b03..c90a20cd5 100644
--- a/integrations/amazon_bedrock/src/amazon_bedrock_haystack/generators/amazon_bedrock.py
+++ b/integrations/amazon_bedrock/src/amazon_bedrock_haystack/generators/amazon_bedrock.py
@@ -6,12 +6,12 @@
 import boto3
 from botocore.exceptions import BotoCoreError, ClientError
 from haystack import component
-from haystack.errors import (
+
+from amazon_bedrock_haystack.errors import (
     AmazonBedrockConfigurationError,
     AmazonBedrockInferenceError,
     AWSConfigurationError,
 )
-
 from amazon_bedrock_haystack.generators.amazon_bedrock_adapters import (
     AI21LabsJurassic2Adapter,
     AmazonTitanAdapter,
diff --git a/integrations/amazon_bedrock/tests/test_amazon_bedrock.py b/integrations/amazon_bedrock/tests/test_amazon_bedrock.py
index cecb28578..1dab863e5 100644
--- a/integrations/amazon_bedrock/tests/test_amazon_bedrock.py
+++ b/integrations/amazon_bedrock/tests/test_amazon_bedrock.py
@@ -3,8 +3,8 @@
 
 import pytest
 from botocore.exceptions import BotoCoreError
-from haystack.errors import AmazonBedrockConfigurationError
 
+from amazon_bedrock_haystack.errors import AmazonBedrockConfigurationError
 from amazon_bedrock_haystack.generators.amazon_bedrock import AmazonBedrockGenerator
 from amazon_bedrock_haystack.generators.amazon_bedrock_adapters import (
     AI21LabsJurassic2Adapter,