diff --git a/packages/amplify-graphql-api-construct-tests/src/__tests__/generations/generation.test.ts b/packages/amplify-graphql-api-construct-tests/src/__tests__/generations/generation.test.ts
index fee2f093e6..6d348310c5 100644
--- a/packages/amplify-graphql-api-construct-tests/src/__tests__/generations/generation.test.ts
+++ b/packages/amplify-graphql-api-construct-tests/src/__tests__/generations/generation.test.ts
@@ -63,7 +63,9 @@ describe('generation', () => {
         };
 
         const generateRecipeResult = await doAppSyncGraphqlQuery({ ...args, query: generateRecipe, variables });
+        expect(generateRecipeResult.body.errors).toBeUndefined();
         const recipe = generateRecipeResult.body.data.generateRecipe;
+        expect(recipe).not.toBeNull();
         expect(recipe.name).toBeDefined();
       });
     });
diff --git a/packages/amplify-graphql-api-construct-tests/src/__tests__/generations/graphql/schema-generation.graphql b/packages/amplify-graphql-api-construct-tests/src/__tests__/generations/graphql/schema-generation.graphql
index 9ee59e600e..244eee03da 100644
--- a/packages/amplify-graphql-api-construct-tests/src/__tests__/generations/graphql/schema-generation.graphql
+++ b/packages/amplify-graphql-api-construct-tests/src/__tests__/generations/graphql/schema-generation.graphql
@@ -5,15 +5,11 @@ type Recipe {
 
 type Query {
   summarize(input: String): String
-    @generation(
-      aiModel: "anthropic.claude-3-haiku-20240307-v1:0"
-      systemPrompt: "summarize the input."
-      inferenceConfiguration: { temperature: 0.5 }
-    )
+    @generation(aiModel: "amazon.nova-lite-v1:0", systemPrompt: "summarize the input.", inferenceConfiguration: { temperature: 0.5 })
 
   generateRecipe(description: String): Recipe
-    @generation(aiModel: "anthropic.claude-3-haiku-20240307-v1:0", systemPrompt: "You are a 3 star michelin chef that generates recipes.")
+    @generation(aiModel: "amazon.nova-lite-v1:0", systemPrompt: "You are a 3 star michelin chef that generates recipes.")
 
   solveEquation(equation: String): Int
-    @generation(aiModel: "anthropic.claude-3-haiku-20240307-v1:0", systemPrompt: "Solve the equation and return the result.")
+    @generation(aiModel: "amazon.nova-lite-v1:0", systemPrompt: "Solve the equation and return the result.")
 }