diff --git a/forward_engineering/helpers/updateHelpers/tableHelper.js b/forward_engineering/helpers/updateHelpers/tableHelper.js
index 3ab774f..b0585d0 100644
--- a/forward_engineering/helpers/updateHelpers/tableHelper.js
+++ b/forward_engineering/helpers/updateHelpers/tableHelper.js
@@ -121,6 +121,12 @@ const hydrateColumn = ({ tableName, keyspaceName, isOldModel, property, udtMap,
 	};
 };
 
+const getTableParameter = (item, key) => {
+	const isCreated = item?.role?.compMod?.created;
+
+	return isCreated ? item?.role?.[key] : item?.role?.compMod?.[key]?.new;
+};
+
 const addToKeysHashType = (keysHash, keys) => {
 	return Object.entries(keysHash).reduce((keysHash, [id, key]) => {
 		const type = (keys.find(key => key.keyId === id) || {}).type;
@@ -211,8 +217,8 @@ const getAddTable = addTableData => {
 	}
 	let table = addTableData.item;
 	const data = addTableData.data;
-	const compositePartitionKey = table?.role?.compMod?.compositePartitionKey?.new || [];
-	const compositeClusteringKey = table?.role?.compMod?.compositeClusteringKey?.new || [];
+	const compositePartitionKey = getTableParameter(table, 'compositePartitionKey') || [];
+	const compositeClusteringKey = getTableParameter(table, 'compositeClusteringKey') || [];
 
 	const entityData = [
 		{