Fixing a logic booboo