small cosmetic update to logic