Fix class attribute