Fixed saveModel