diff --git a/app/server/appsmith-server/src/main/java/com/appsmith/server/services/NewActionServiceImpl.java b/app/server/appsmith-server/src/main/java/com/appsmith/server/services/NewActionServiceImpl.java
index 37dd7df91d..c53e44a1af 100644
--- a/app/server/appsmith-server/src/main/java/com/appsmith/server/services/NewActionServiceImpl.java
+++ b/app/server/appsmith-server/src/main/java/com/appsmith/server/services/NewActionServiceImpl.java
@@ -684,6 +684,11 @@ public class NewActionServiceImpl extends BaseService<NewActionRepository, NewAc
     }
 
     private ActionExecutionResult addDataTypesAndSetSuggestedWidget(ActionExecutionResult result, Boolean viewMode) {
+
+        if(FALSE.equals(viewMode)) {
+            result.setSuggestedWidgets(getSuggestedWidget(result.getBody()));
+        }
+
         /*
          * - Do not process if data types are already present.
          * - It means that data types have been added by specific plugin.
@@ -695,10 +700,6 @@ public class NewActionServiceImpl extends BaseService<NewActionRepository, NewAc
 
         result.setDataTypes(getDisplayDataTypes(result.getBody()));
 
-        if(FALSE.equals(viewMode)) {
-            result.setSuggestedWidgets(getSuggestedWidget(result.getBody()));
-        }
-
         return result;
     }