diff --git a/src/Appwrite/Migration/Version/V12.php b/src/Appwrite/Migration/Version/V12.php
index d7b279461c..6bba8b11f5 100644
--- a/src/Appwrite/Migration/Version/V12.php
+++ b/src/Appwrite/Migration/Version/V12.php
@@ -216,6 +216,7 @@ class V12 extends Migration
                                 $nextDocument = null;
                             } else {
                                 $nextDocument = end($documents);
+                                $nextDocument->setAttribute('$collection', 'files');
                             }
                         } while (!is_null($nextDocument));