I first used the dict that is returned when you just create the model, and there's no model_fields there yet. That's why I check if status is FINISHED