Typo + slight code change