diff --git a/sdk/servicebus/service-bus/test/public/atomManagement.spec.ts b/sdk/servicebus/service-bus/test/public/atomManagement.spec.ts
index 29f89999c23a..f2e400a929e9 100644
--- a/sdk/servicebus/service-bus/test/public/atomManagement.spec.ts
+++ b/sdk/servicebus/service-bus/test/public/atomManagement.spec.ts
@@ -1374,6 +1374,7 @@ describe("ATOM APIs", () => {
         enablePartitioning: false,
         enableExpress: false,
         maxSizeInMegabytes: 1024,
+        maxMessageSizeInKilobytes: 256,
         userMetadata: undefined,
         requiresDuplicateDetection: false,
         status: "Active",
@@ -1407,6 +1408,7 @@ describe("ATOM APIs", () => {
         enablePartitioning: true,
         enableExpress: false,
         maxSizeInMegabytes: 16384,
+        maxMessageSizeInKilobytes: 256,
         autoDeleteOnIdle: "P10675199DT2H48M5.4775807S",
         authorizationRules: undefined,
         userMetadata: "test metadata",
@@ -1663,6 +1665,7 @@ describe("ATOM APIs", () => {
         lockDuration: "PT1M",
         maxDeliveryCount: 10,
         maxSizeInMegabytes: 1024,
+        maxMessageSizeInKilobytes: 256,
         name: managementQueue1,
         requiresDuplicateDetection: false,
         requiresSession: false,
@@ -1735,6 +1738,7 @@ describe("ATOM APIs", () => {
         enablePartitioning: true,
         enableExpress: false,
         maxSizeInMegabytes: 16384,
+        maxMessageSizeInKilobytes: 256,
         forwardDeadLetteredMessagesTo: undefined,
         forwardTo: undefined,
         userMetadata: "test metadata",
@@ -2037,6 +2041,7 @@ describe("ATOM APIs", () => {
         ],
         maxDeliveryCount: 5,
         maxSizeInMegabytes: 16384,
+        maxMessageSizeInKilobytes: 256,
         autoDeleteOnIdle: "PT2H",
         forwardDeadLetteredMessagesTo: undefined,
         forwardTo: undefined,
@@ -2234,6 +2239,7 @@ describe("ATOM APIs", () => {
         autoDeleteOnIdle: "PT2H",
         supportOrdering: true,
         maxSizeInMegabytes: 3072,
+        maxMessageSizeInKilobytes: 256,
         enableBatchedOperations: true,
         enablePartitioning: false,
         enableExpress: false,