Fixed import in collection_mock.py
Fixed import in collection_mock.py