Update class name from old mod