Two small fixes in models