diff --git a/PowerSync/PowerSync.Common/MDSQLite/MDSQLiteAdapter.cs b/PowerSync/PowerSync.Common/MDSQLite/MDSQLiteAdapter.cs
index 5ae03cf..9d71447 100644
--- a/PowerSync/PowerSync.Common/MDSQLite/MDSQLiteAdapter.cs
+++ b/PowerSync/PowerSync.Common/MDSQLite/MDSQLiteAdapter.cs
@@ -248,10 +248,8 @@ public async Task WriteLock(Func<ILockContext, Task> fn, DBLockOptions? options
         using (await writeMutex.LockAsync())
         {
             await fn(writeConnection);
+            writeConnection.FlushUpdates();
         }
-
-        writeConnection.FlushUpdates();
-
     }
 
     public async Task<T> WriteLock<T>(Func<ILockContext, Task<T>> fn, DBLockOptions? options = null)
@@ -262,10 +260,9 @@ public async Task<T> WriteLock<T>(Func<ILockContext, Task<T>> fn, DBLockOptions?
         using (await writeMutex.LockAsync())
         {
             result = await fn(writeConnection);
+            writeConnection.FlushUpdates();
         }
 
-        writeConnection.FlushUpdates();
-
         return result;
     }