diff --git a/RevenueCatUI/Purchasing/PurchaseHandler+TestData.swift b/RevenueCatUI/Purchasing/PurchaseHandler+TestData.swift
index 5d9f2f21d7..b88004dd80 100644
--- a/RevenueCatUI/Purchasing/PurchaseHandler+TestData.swift
+++ b/RevenueCatUI/Purchasing/PurchaseHandler+TestData.swift
@@ -19,7 +19,7 @@ import RevenueCat
 @available(iOS 15.0, macOS 12.0, tvOS 15.0, *)
 extension PurchaseHandler {
 
-    static func mock(customerInfo: CustomerInfo = TestData.customerInfo) -> Self {
+    static func mock(_ customerInfo: CustomerInfo = TestData.customerInfo) -> Self {
         return self.init(
             purchases: MockPurchases { _ in
                 return (
diff --git a/Tests/RevenueCatUITests/Purchasing/PurchaseHandlerTests.swift b/Tests/RevenueCatUITests/Purchasing/PurchaseHandlerTests.swift
index aced885f44..dbc8d0032c 100644
--- a/Tests/RevenueCatUITests/Purchasing/PurchaseHandlerTests.swift
+++ b/Tests/RevenueCatUITests/Purchasing/PurchaseHandlerTests.swift
@@ -66,7 +66,7 @@ class PurchaseHandlerTests: TestCase {
     }
 
     func testRestorePurchasesWithActiveSubscriptions() async throws {
-        let handler: PurchaseHandler = .mock(customerInfo: Self.customerInfoWithSubscriptions)
+        let handler: PurchaseHandler = .mock(Self.customerInfoWithSubscriptions)
 
         let result = try await handler.restorePurchases()
         expect(result.info) === Self.customerInfoWithSubscriptions
@@ -74,7 +74,7 @@ class PurchaseHandlerTests: TestCase {
     }
 
     func testRestorePurchasesWithNonSubscriptions() async throws {
-        let handler: PurchaseHandler = .mock(customerInfo: Self.customerInfoWithNonSubscriptions)
+        let handler: PurchaseHandler = .mock(Self.customerInfoWithNonSubscriptions)
 
         let result = try await handler.restorePurchases()
         expect(result.info) === Self.customerInfoWithNonSubscriptions