Error with class name fixed