small model fixes