fix the final model save bug