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