Tweak the infinite loop code.