Minor logic tweak.