@Override
    public void visitForStatement(@NotNull PsiForStatement statement) {
      expectedType = PsiTypes.booleanType();
    }
@Override
    public void visitForStatement(@NotNull PsiForStatement statement)