lib/engineyard-serverside/servers.rb in engineyard-serverside-2.0.0.rc1 vs lib/engineyard-serverside/servers.rb in engineyard-serverside-2.0.0

@@ -49,10 +49,10 @@
       # Cache them so we don't have to find them every time
       # Accepts a block (because it's confusing when you send a block to this
       # method and it doesn't yield and it's better than raising)
       def roles(*select_roles, &block)
         if block_given?
-          return yield roles(*select_roles)
+          return yield(roles(*select_roles))
         end
 
         roles_set = Set.new select_roles.flatten.compact.map{|r| r.to_sym}
         if roles_set.empty? || roles_set.include?(:all)
           self