Fix class name bug