Fixed a typo left after I changed the class name