Getting rid of TaskStatus.get_collection uses and replacing with mongoengine queries