diff --git a/server/pkg/repo/storagebonus/referral_codes_test.go b/server/pkg/repo/storagebonus/referral_codes_test.go
index 379d932334..117dfb1352 100644
--- a/server/pkg/repo/storagebonus/referral_codes_test.go
+++ b/server/pkg/repo/storagebonus/referral_codes_test.go
@@ -65,7 +65,7 @@ func TestAddNewReferralCode(t *testing.T) {
 	assert.Nil(t, err)
 
 	newCode := "C22222"
-	err = repo.AddNewCode(context.Background(), userID, newCode)
+	err = repo.AddNewCode(context.Background(), userID, newCode, false)
 	assert.Nil(t, err)
 
 	referralCode, err := repo.GetCode(context.Background(), userID)