Updated the class to the correct name