diff --git a/ingestion/src/metadata/ingestion/lineage/sql_lineage.py b/ingestion/src/metadata/ingestion/lineage/sql_lineage.py index 7851a1320eb..4cae5a53948 100644 --- a/ingestion/src/metadata/ingestion/lineage/sql_lineage.py +++ b/ingestion/src/metadata/ingestion/lineage/sql_lineage.py @@ -176,19 +176,8 @@ def get_table_entities_from_query( table_entities = search_table_entities( metadata=metadata, service_name=service_name, - database=database_query, - database_schema=schema_query, - table=table, - ) - - if table_entities: - return table_entities - - table_entities = search_table_entities( - metadata=metadata, - service_name=service_name, - database=database_name, - database_schema=database_schema, + database=database_query if database_query else database_name, + database_schema=schema_query if schema_query else database_schema, table=table, )