Minor change to fix saving of models.