Fix setClazz check