small fixes to models