fix(e2e): account tests

matter-labs · Apr 23, 2024 · a1d1b4e · a1d1b4e
1 parent b9d74a2
commit a1d1b4e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/api/src/balance/balance.entity.ts b/packages/api/src/balance/balance.entity.ts
@@ -25,7 +25,10 @@ export class Balance extends BaseEntity {
 
   @AfterLoad()
   populateBaseToken() {
-    if (!this.token && this.tokenAddress.toLowerCase() === baseTokenData.l2Address.toLowerCase()) {
+    if (
+      !this.token &&
+      (this.tokenAddress === undefined || this.tokenAddress.toLowerCase() === baseTokenData.l2Address.toLowerCase())
+    ) {
       this.token = baseTokenData as Token;
     }
   }