Fixed incorrect class name

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