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