diff --git a/forward_engineering/dbtProvider.js b/forward_engineering/dbtProvider.js
index f66c6fb..a9171c6 100644
--- a/forward_engineering/dbtProvider.js
+++ b/forward_engineering/dbtProvider.js
@@ -80,6 +80,17 @@ class DbtProvider {
 	getColumnConstraints({ columnDefinition, jsonSchema }) {
 		return getColumnConstraints({ columnDefinition });
 	}
+
+	/**
+	 * @param {{ modelData: object[]; containerData: object[]; entityData: object[];}}
+	 * @returns {{ databaseName?: string, schemaName?: string }}
+	 */
+	getEntityProperties({ modelData, containerData, entityData }) {
+		return {
+			databaseName: containerData?.[0]?.databaseName,
+			schemaName: containerData?.[0]?.code ?? containerData?.[0]?.name,
+		};
+	}
 }
 
 module.exports = DbtProvider;