Added more import mocks