diff --git a/app/client/src/workers/Evaluation/JSObject/utils.ts b/app/client/src/workers/Evaluation/JSObject/utils.ts
index 1db35c0abf..7be11c15cb 100644
--- a/app/client/src/workers/Evaluation/JSObject/utils.ts
+++ b/app/client/src/workers/Evaluation/JSObject/utils.ts
@@ -8,7 +8,7 @@ import { ParsedBody, ParsedJSSubAction } from "utils/JSPaneUtils";
 import { unset, set, get } from "lodash";
 import { BatchedJSExecutionData } from "reducers/entityReducers/jsActionsReducer";
 import { select } from "redux-saga/effects";
-import { AppState } from "ce/reducers";
+import { AppState } from "@appsmith/reducers";
 import { JSAction } from "entities/JSCollection";
 import { getJSFunctionFromName } from "selectors/entitiesSelector";
 import { isJSAction } from "@appsmith/workers/Evaluation/evaluationUtils";
@@ -264,6 +264,7 @@ export function* sortJSExecutionDataByCollectionId(
     const jsAction: JSAction | undefined = yield select((state: AppState) =>
       getJSFunctionFromName(state, jsfuncFullName),
     );
+
     if (jsAction && jsAction.collectionId) {
       if (sortedData[jsAction.collectionId]) {
         sortedData[jsAction.collectionId].push({