diff --git a/tools/migration/issue_882.js b/tools/migration/issue_882.js
index 67a1551e0..8dab9bb43 100644
--- a/tools/migration/issue_882.js
+++ b/tools/migration/issue_882.js
@@ -17,12 +17,12 @@ const migrate = async (doc) => {
 async function main() {
 	let i = 0;
 
-	const count = await db.get('drive_files').count({});
+	const count = await DriveFile.count({});
 
 	const iterate = async () => {
 		if (i == count) return true;
 		console.log(`${i} / ${count}`);
-		const doc = (await db.get('drive_files').find({}, { limit: 1, skip: i }))[0]
+		const doc = (await DriveFile.find({}, { limit: 1, skip: i }))[0]
 		const res = await migrate(doc);
 		if (!res) {
 			return false;