Bug in set referral code fixed