Typo fix

- Change class name on new source code