diff --git a/.circleci/config.base.yml b/.circleci/config.base.yml
index 5698219173a..6ad6fb1d403 100644
--- a/.circleci/config.base.yml
+++ b/.circleci/config.base.yml
@@ -287,7 +287,8 @@ jobs:
 
   amplify_migration_tests_non_multi_env_layers:
     <<: *defaults
-
+    environment:
+      AMPLIFY_PATH: /home/circleci/.npm-global/lib/node_modules/@aws-amplify/cli/bin/amplify
     steps:
       - attach_workspace:
           at: ./
@@ -309,7 +310,8 @@ jobs:
 
   amplify_migration_tests_multi_env_layers:
     <<: *defaults
-
+    environment:
+      AMPLIFY_PATH: /home/circleci/.npm-global/lib/node_modules/@aws-amplify/cli/bin/amplify
     steps:
       - attach_workspace:
           at: ./
diff --git a/.circleci/config.yml b/.circleci/config.yml
index 038dcdede0d..d9621ee5666 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -297,6 +297,8 @@ jobs:
     working_directory: ~/repo
     docker: *ref_1
     resource_class: large
+    environment:
+      AMPLIFY_PATH: /home/circleci/.npm-global/lib/node_modules/@aws-amplify/cli/bin/amplify
     steps:
       - attach_workspace:
           at: ./
@@ -322,6 +324,8 @@ jobs:
     working_directory: ~/repo
     docker: *ref_1
     resource_class: large
+    environment:
+      AMPLIFY_PATH: /home/circleci/.npm-global/lib/node_modules/@aws-amplify/cli/bin/amplify
     steps:
       - attach_workspace:
           at: ./
diff --git a/packages/amplify-e2e-tests/src/__tests__/function_4.test.ts b/packages/amplify-e2e-tests/src/__tests__/function_4.test.ts
index 2e40fcf61d8..6560a53b647 100644
--- a/packages/amplify-e2e-tests/src/__tests__/function_4.test.ts
+++ b/packages/amplify-e2e-tests/src/__tests__/function_4.test.ts
@@ -314,6 +314,6 @@ describe('add function with layers for runtime python', () => {
     const payload = '{}';
     const response = await functionCloudInvoke(projRoot, { funcName: functionName, payload: payload });
 
-    expect(JSON.parse(response.Payload.toString()).body).toEqual(helloWorldSuccessOutput);
+    expect(JSON.parse(response.Payload.toString()).body).toMatch(helloWorldSuccessOutput);
   });
 });