diff --git a/src/Appwrite/Databases/TransactionState.php b/src/Appwrite/Databases/TransactionState.php
index 1caae8c629..495a7b70ef 100644
--- a/src/Appwrite/Databases/TransactionState.php
+++ b/src/Appwrite/Databases/TransactionState.php
@@ -49,15 +49,15 @@ class TransactionState
 
             switch ($action) {
                 case 'create':
-                    if ($documentId) {
-                        $state[$collectionId][$documentId] = [
+                    $docId = $documentId ?? ($data['$id'] ?? null);
+                    if ($docId) {
+                        $state[$collectionId][$docId] = [
                             'action' => 'create',
                             'document' => new Document($data),
                             'exists' => true
                         ];
                     }
                     break;
-
                 case 'update':
                     if (isset($state[$collectionId][$documentId])) {
                         // Update existing document in transaction state