Tweak some code.