lib/yard/handlers/base.rb in yard-0.8.1 vs lib/yard/handlers/base.rb in yard-0.8.2

@@ -354,15 +354,16 @@
       # @yield a block to execute with the given state values.
       def push_state(opts = {}, &block)
         opts = {
           :namespace => namespace,
           :scope => :instance,
-          :owner => owner || namespace
+          :owner => owner || namespace,
+          :visibility => nil
         }.update(opts)
 
         ns, vis, sc, oo = namespace, visibility, scope, owner
         self.namespace = opts[:namespace]
-        self.visibility = :public
+        self.visibility = opts[:visibility] || :public
         self.scope = opts[:scope]
         self.owner = opts[:owner]
 
         yield
 
\ No newline at end of file