fixed class name problem