A typo was fixed
Slight code refactoring