Fixed class name error