diff --git a/modules/test-utils/src/generate-layer-tests.js b/modules/test-utils/src/generate-layer-tests.js
index 512e7f4bb7b..92381e6448b 100644
--- a/modules/test-utils/src/generate-layer-tests.js
+++ b/modules/test-utils/src/generate-layer-tests.js
@@ -98,9 +98,10 @@ export function generateLayerTests({
   testCases.forEach(testCase => {
     testCase.title = wrapTestCaseTitle(testCase.title);
     testCase.onBeforeUpdate = onBeforeUpdate;
-    testCases.onAfterUpdate = _onAfterUpdate;
   });
 
+  testCases[2].onAfterUpdate = _onAfterUpdate;
+
   return testCases;
 }