Added mocks for test