Update mocked classes for test