@Override
    public void visitWhileStatement(@NotNull PsiWhileStatement whileStatement) {
      expectedType = PsiTypes.booleanType();
    }
@Override
    public void visitWhileStatement(@NotNull PsiWhileStatement whileStatement)