diff --git a/tools/migration/use-gridfs.js b/tools/migration/use-gridfs.js
index bb7070f55..a9d2b12e9 100644
--- a/tools/migration/use-gridfs.js
+++ b/tools/migration/use-gridfs.js
@@ -46,7 +46,7 @@ const migrateToGridFS = async (doc) => {
 }
 
 async function main() {
-	const count = await DriveFile.count({});
+	const count = await db.get('drive_files').count({});
 
 	console.log(`there are ${count} files.`)