Fixed bug in collection_mock.py