diff --git a/ruby/lib/jam_ruby/models/affiliate_partner.rb b/ruby/lib/jam_ruby/models/affiliate_partner.rb
index 46906fbde..6df9dc1a2 100644
--- a/ruby/lib/jam_ruby/models/affiliate_partner.rb
+++ b/ruby/lib/jam_ruby/models/affiliate_partner.rb
@@ -33,11 +33,12 @@ class JamRuby::AffiliatePartner < ActiveRecord::Base
   end
 
   def referrals_by_date
-    User.where(:affiliate_referral_id => self.id)
+    by_date = User.where(:affiliate_referral_id => self.id)
       .group('DATE(created_at)')
       .having("COUNT(*) > 0")
       .order('date_created_at DESC')
       .count
+    block_given? ? yield(by_date) : by_date
   end
 
 end