don't try add_extensions on plain objects.

despo · Jan 17, 2011 · 6e1c44f · 6e1c44f
1 parent 7163913
commit 6e1c44f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lib/pacer/core/route.rb b/lib/pacer/core/route.rb
@@ -113,8 +113,11 @@ def each_element
         else
           if block_given?
             while item = iter.next
-              item.graph ||= g if g and item.respond_to? :graph=
-              yield item.add_extensions(extensions)
+              if item.respond_to? :graph=
+                item.graph ||= g if g and item.respond_to? :graph=
+                item = item.add_extensions(extensions)
+              end
+              yield item
             end
           else
             iter.extend IteratorExtensionsMixin