Fix slipup when cleaning code