1 files changed, 9 insertions, 9 deletions

diff --git a/pbt/ts/src/bank.test.ts b/pbt/ts/src/bank.test.ts
index f6aa7b0..5939034 100644
--- a/pbt/ts/src/bank.test.ts
+++ b/pbt/ts/src/bank.test.ts
@@ -61,20 +61,20 @@ const genOperations = (accounts: Map<Account, number>) =>
 
 const correctBalance =
   (initialAccounts: Map<Account, number>) =>
-  (ops: Operation[]): Promise<boolean> => {
+  async (ops: Operation[]): Promise<boolean> => {
     const expectedBalances = new Map(initialAccounts);
     const api = new SimpleBank(initialAccounts);
 
     for (const op of ops) {
       switch (op.op) {
         case 'GetBalance': {
-          const retrieved = toNullable(api.balance(op.account));
+          const retrieved = toNullable(await api.balance(op.account));
           const expected = expectedBalances.get(op.account);
           if (retrieved !== expected) {
             console.log(
               `Balance mismatch for account ${op.account}: expected ${expected}, got ${retrieved}`,
             );
-            return Promise.resolve(false);
+            return false;
           }
           break;
         }
@@ -95,24 +95,24 @@ const correctBalance =
 
     // finally check all balances
     for (const [account, expectedBal] of expectedBalances.entries()) {
-      const retrieved = toNullable(api.balance(account));
+      const retrieved = toNullable(await api.balance(account));
       if (retrieved !== expectedBal) {
         console.log(
           `Final balance mismatch for account ${account}: expected ${expectedBal}, got ${retrieved}`,
         );
-        return Promise.resolve(false);
+        return false;
       }
     }
 
-    return Promise.resolve(true);
+    return true;
   };
 
 describe('Bank Simulator', () => {
   test('can retrieve balance given account exists', async () => {
-    const prop = forall(genAccounts, accounts => {
+    const prop = forall(genAccounts, async accounts => {
       const api = new SimpleBank(accounts);
-      return property(keyValueOf(accounts), ([accountNr, bal]) => {
-        const retrieved = toNullable(api.balance(accountNr));
+      return property(keyValueOf(accounts), async ([accountNr, bal]) => {
+        const retrieved = toNullable(await api.balance(accountNr));
         return Promise.resolve(retrieved === bal);
       });
     });