Small typo/logic fix