Small change to collision code