lib/grumlin/shortcut_proxy.rb in grumlin-0.14.0 vs lib/grumlin/shortcut_proxy.rb in grumlin-0.14.1

@@ -41,10 +41,12 @@
     end
 
     private
 
     def wrap_result(result)
-      return self.class.new(result, @shortcuts, parent: @parent) if result.is_a?(AnonymousStep)
+      if result.is_a?(AnonymousStep) || result.is_a?(Traversal)
+        return self.class.new(result, @shortcuts, parent: @parent)
+      end
 
       result
     end
   end
 end