Fix class check.