1 files changed, 9 insertions, 9 deletions

diff --git a/pbt/ts/src/bank.test.ts b/pbt/ts/src/bank.test.ts
index d503faf..f6aa7b0 100644
--- a/pbt/ts/src/bank.test.ts
+++ b/pbt/ts/src/bank.test.ts
@@ -61,7 +61,7 @@ const genOperations = (accounts: Map<Account, number>) =>
 
 const correctBalance =
   (initialAccounts: Map<Account, number>) =>
-  (ops: Operation[]): boolean => {
+  (ops: Operation[]): Promise<boolean> => {
     const expectedBalances = new Map(initialAccounts);
     const api = new SimpleBank(initialAccounts);
 
@@ -74,7 +74,7 @@ const correctBalance =
             console.log(
               `Balance mismatch for account ${op.account}: expected ${expected}, got ${retrieved}`,
             );
-            return false;
+            return Promise.resolve(false);
           }
           break;
         }
@@ -100,34 +100,34 @@ const correctBalance =
         console.log(
           `Final balance mismatch for account ${account}: expected ${expectedBal}, got ${retrieved}`,
         );
-        return false;
+        return Promise.resolve(false);
       }
     }
 
-    return true;
+    return Promise.resolve(true);
   };
 
 describe('Bank Simulator', () => {
-  test('can retrieve balance given account exists', () => {
+  test('can retrieve balance given account exists', async () => {
     const prop = forall(genAccounts, accounts => {
       const api = new SimpleBank(accounts);
       return property(keyValueOf(accounts), ([accountNr, bal]) => {
         const retrieved = toNullable(api.balance(accountNr));
-        return retrieved === bal;
+        return Promise.resolve(retrieved === bal);
       });
     });
 
-    const result = Property.run(prop);
+    const result = await Property.run(prop);
 
     expect(result.result).toBe('OK');
   });
 
-  test('correctly update balance given a sequence of deposits and withdrawals', () => {
+  test('correctly update balance given a sequence of deposits and withdrawals', async () => {
     const prop = forall(genAccounts, accounts => {
       return property(genOperations(accounts), correctBalance(accounts));
     });
 
-    const result = Property.run(prop);
+    const result = await Property.run(prop);
 
     expect(result.result).toBe('OK');
   });