Removed unecessary handling. Changed test_collection.py