Fixed incorrect class name

Updated the class name to fix the issue mentioned in 138.