diff --git a/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/invoke.json b/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/invoke.json
index a6234eedaf..fb9b4df449 100644
--- a/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/invoke.json
+++ b/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/invoke.json
@@ -1,12 +1,12 @@
 {
   "identifier": "INVOKE_FUNCTION",
-  "controlType": "SECTION",
+  "controlType": "SECTION_V2",
   "conditionals": {
     "show": "{{actionConfiguration.formData.command.data === 'INVOKE_FUNCTION'}}"
   },
   "children": [
     {
-      "controlType": "SECTION",
+      "controlType": "DOUBLE_COLUMN_ZONE",
       "label": "Details of lambda function",
       "children": [
         {
@@ -63,7 +63,12 @@
           "fetchOptionsConditionally": false,
           "setFirstOptionAsDefault": true,
           "alternateViewTypes": ["json"]
-        },
+        }
+      ]
+    },
+    {
+      "controlType": "SINGLE_COLUMN_ZONE",
+      "children": [
         {
           "label": "Post body",
           "configProperty": "actionConfiguration.formData.body.data",
diff --git a/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/list.json b/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/list.json
index 72e004a0c0..897468ad1d 100644
--- a/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/list.json
+++ b/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/list.json
@@ -1,6 +1,6 @@
 {
   "identifier": "LIST_FUNCTIONS",
-  "controlType": "SECTION",
+  "controlType": "SECTION_V2",
   "conditionals": {
     "show": "{{actionConfiguration.formData.command.data === 'LIST_FUNCTIONS'}}"
   },
diff --git a/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/root.json b/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/root.json
index 5eda374173..00bc0a8e40 100644
--- a/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/root.json
+++ b/app/server/appsmith-plugins/awsLambdaPlugin/src/main/resources/editor/root.json
@@ -1,23 +1,29 @@
 {
   "editor": [
     {
-      "controlType": "SECTION",
+      "controlType": "SECTION_V2",
       "identifier": "SELECTOR",
+
       "children": [
         {
-          "label": "Command",
-          "description": "Choose the method you would like to use",
-          "configProperty": "actionConfiguration.formData.command.data",
-          "controlType": "DROP_DOWN",
-          "initialValue": "LIST_FUNCTIONS",
-          "options": [
+          "controlType": "DOUBLE_COLUMN_ZONE",
+          "children": [
             {
-              "label": "List all functions",
-              "value": "LIST_FUNCTIONS"
-            },
-            {
-              "label": "Invoke a function",
-              "value": "INVOKE_FUNCTION"
+              "label": "Command",
+              "description": "Choose the method you would like to use",
+              "configProperty": "actionConfiguration.formData.command.data",
+              "controlType": "DROP_DOWN",
+              "initialValue": "LIST_FUNCTIONS",
+              "options": [
+                {
+                  "label": "List all functions",
+                  "value": "LIST_FUNCTIONS"
+                },
+                {
+                  "label": "Invoke a function",
+                  "value": "INVOKE_FUNCTION"
+                }
+              ]
             }
           ]
         }