Minor fix on comments
- fixing typo on class