This doesn't apply without `return_models`