Bit of code cleanup.