Possible bug in model_save.py