diff --git a/forward_engineering/generateContainerScript.js b/forward_engineering/generateContainerScript.js
index 72352cf..7de632f 100644
--- a/forward_engineering/generateContainerScript.js
+++ b/forward_engineering/generateContainerScript.js
@@ -19,9 +19,11 @@ const generateContainerScript = (data, logger, callback, app) => {
 			),
 		);
 		if (data.options?.targetScriptOptions?.keyword === 'containerSettingsJson') {
+			const partitionKeys = _.get(data.containerData, '[0].partitionKey', []);
 			const uniqueKeys = _.get(data.containerData, '[0].uniqueKey', []);
+
 			const scriptData = {
-				partitionKey: getPartitionKey(data.containerData),
+				...(partitionKeys.length && { partitionKey: getPartitionKey(data.containerData) }),
 				...(uniqueKeys.length && getUniqueKeyPolicyScript(uniqueKeys)),
 				indexingPolicy: getIndexPolicyScript(data.containerData),
 				...(withSamples && { sample: samples }),
diff --git a/forward_engineering/generateScript.js b/forward_engineering/generateScript.js
index c71a27f..bf0240e 100644
--- a/forward_engineering/generateScript.js
+++ b/forward_engineering/generateScript.js
@@ -7,10 +7,11 @@ const { updateSample } = require('./helpers/updateSample');
 
 const generateScript = (data, logger, callback, app) => {
 	try {
+		const partitionKeys = _.get(data.containerData, '[0].partitionKey', []);
 		const uniqueKeys = _.get(data.containerData, '[0].uniqueKey', []);
 
 		const script = {
-			partitionKey: getPartitionKey(data.containerData),
+			...(partitionKeys.length && { partitionKey: getPartitionKey(data.containerData) }),
 			indexingPolicy: getIndexPolicyScript(data.containerData),
 			...(uniqueKeys.length && getUniqueKeyPolicyScript(uniqueKeys)),
 			sample: updateSample(JSON.parse(data.jsonData), data.containerData[0], data.entityData[0]),