diff --git a/engine/src/flutter/impeller/aiks/testing/context_mock.h b/engine/src/flutter/impeller/aiks/testing/context_mock.h
index 25283e7498e..2650105de4a 100644
--- a/engine/src/flutter/impeller/aiks/testing/context_mock.h
+++ b/engine/src/flutter/impeller/aiks/testing/context_mock.h
@@ -80,6 +80,8 @@ class ContextMock : public Context {
   MOCK_CONST_METHOD0(GetPipelineLibrary, std::shared_ptr<PipelineLibrary>());
 
   MOCK_CONST_METHOD0(CreateCommandBuffer, std::shared_ptr<CommandBuffer>());
+
+  MOCK_METHOD0(Shutdown, void());
 };
 
 }  // namespace testing