diff --git a/ingestion/src/metadata/ingestion/source/dynamodb.py b/ingestion/src/metadata/ingestion/source/dynamodb.py
index 826d1034e30..47e2eaa16cb 100644
--- a/ingestion/src/metadata/ingestion/source/dynamodb.py
+++ b/ingestion/src/metadata/ingestion/source/dynamodb.py
@@ -4,6 +4,7 @@ import uuid
 from typing import Iterable
 
 from metadata.generated.schema.entity.data.database import Database
+from metadata.generated.schema.entity.data.databaseSchema import DatabaseSchema
 from metadata.generated.schema.entity.data.table import Column, Table
 from metadata.generated.schema.entity.services.connections.database.dynamoDBConnection import (
     DynamoDBConnection,