Add queryset param to get_object method to fix tests