Fixed small bug in model