Updating mocks for imports