Add required import of get_collection.