Fix class in formatting