fix model save