Remove unnecessary mocks