small bug in teh model