Fixed small bug in HIT model extend