Wrong class name

Updated the class name