Save models bug fix.