From 94e31d4685a6a3580e6a17a3227adf6f13d5c8c6 Mon Sep 17 00:00:00 2001
From: Sijawusz Pur Rahnama <sija@sija.pl>
Date: Thu, 8 Jun 2023 04:28:37 +0200
Subject: [PATCH] Do the same in `NodeVisitor`

---
 src/ameba/ast/visitors/node_visitor.cr | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/ameba/ast/visitors/node_visitor.cr b/src/ameba/ast/visitors/node_visitor.cr
index c537b86c..315d1b7b 100644
--- a/src/ameba/ast/visitors/node_visitor.cr
+++ b/src/ameba/ast/visitors/node_visitor.cr
@@ -43,6 +43,11 @@ module Ameba::AST
       super @rule, @source
     end
 
+    def visit(node : Crystal::VisibilityModifier)
+      node.exp.visibility = node.modifier
+      true
+    end
+
     {% for name in NODES %}
       # A visit callback for `Crystal::{{ name }}` node.
       #