Updated for models_utils 2.0.2