Fixed class name bug