minor: correcting logic