From ce39bc71b769db4100373922f23599c6711f8c0d Mon Sep 17 00:00:00 2001
From: Erdi Maden <erdi.maden@coinbase.com>
Date: Mon, 20 May 2024 16:13:27 -0500
Subject: [PATCH] Updating Wallet model usage

---
 src/coinbase/user.ts   | 1 -
 src/coinbase/wallet.ts | 4 ++--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/coinbase/user.ts b/src/coinbase/user.ts
index 803bbd1d..593dde4a 100644
--- a/src/coinbase/user.ts
+++ b/src/coinbase/user.ts
@@ -1,5 +1,4 @@
 import { User as UserModel } from "./../client/api";
-import { Coinbase } from "./coinbase";
 import { Wallet } from "./wallet";
 
 /**
diff --git a/src/coinbase/wallet.ts b/src/coinbase/wallet.ts
index 23cddb82..07cf38c0 100644
--- a/src/coinbase/wallet.ts
+++ b/src/coinbase/wallet.ts
@@ -41,6 +41,7 @@ export class Wallet {
    * Instead, use User.createWallet.
    *
    * @constructs Wallet
+   * @throws {ArgumentError} If the model or client is not provided.
    * @throws {InternalError} - If address derivation or caching fails.
    * @throws {APIError} - If the request fails.
    * @returns A promise that resolves with the new Wallet object.
@@ -52,7 +53,7 @@ export class Wallet {
       },
     });
 
-    const wallet = await Wallet.init(walletData.data!);
+    const wallet = await Wallet.init(walletData.data);
 
     await wallet.createAddress();
     await wallet.reload();
@@ -80,7 +81,6 @@ export class Wallet {
     if (!model) {
       throw new ArgumentError("Wallet model cannot be empty");
     }
-
     if (!seed) {
       seed = bip39.generateMnemonic();
     }