From cb78ae80ddceb008a5866c9d37e1fd9895cba2ec Mon Sep 17 00:00:00 2001
From: nsingh-branch <nipun.singh@branch.io>
Date: Thu, 27 Jun 2024 17:36:03 -0700
Subject: [PATCH] Fixed old test

---
 Branch-TestBed/Branch-SDK-Tests/BranchClassTests.m | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/Branch-TestBed/Branch-SDK-Tests/BranchClassTests.m b/Branch-TestBed/Branch-SDK-Tests/BranchClassTests.m
index 447034fa6..d1e7713f1 100644
--- a/Branch-TestBed/Branch-SDK-Tests/BranchClassTests.m
+++ b/Branch-TestBed/Branch-SDK-Tests/BranchClassTests.m
@@ -243,19 +243,20 @@ - (void)testSetDMAParamsForEEA {
 
 - (void)testSetConsumerProtectionAttributionLevel {
     // Set to Reduced and check
-    [Branch setConsumerProtectionAttributionLevel:BranchAttributionLevelReduced];
+    Branch *branch = [Branch getInstance];
+    [branch setConsumerProtectionAttributionLevel:BranchAttributionLevelReduced];
     XCTAssertEqual([BNCPreferenceHelper sharedInstance].attributionLevel, BranchAttributionLevelReduced);
     
     // Set to Minimal and check
-    [Branch setConsumerProtectionAttributionLevel:BranchAttributionLevelMinimal];
+    [branch setConsumerProtectionAttributionLevel:BranchAttributionLevelMinimal];
     XCTAssertEqual([BNCPreferenceHelper sharedInstance].attributionLevel, BranchAttributionLevelMinimal);
     
     // Set to None and check
-    [Branch setConsumerProtectionAttributionLevel:BranchAttributionLevelNone];
+    [branch setConsumerProtectionAttributionLevel:BranchAttributionLevelNone];
     XCTAssertEqual([BNCPreferenceHelper sharedInstance].attributionLevel, BranchAttributionLevelNone);
     
     // Set to Full and check
-    [Branch setConsumerProtectionAttributionLevel:BranchAttributionLevelFull];
+    [branch setConsumerProtectionAttributionLevel:BranchAttributionLevelFull];
     XCTAssertEqual([BNCPreferenceHelper sharedInstance].attributionLevel, BranchAttributionLevelFull);
     
 }