Update mock implementation.