Remove UI mocks