Minor logic fixup.