Fix class restriction