Fix saveModel