diff --git a/tools/migration/node.1510016282.change-gridfs-metadata-name-to-filename.js b/tools/migration/node.1510016282.change-gridfs-metadata-name-to-filename.js
index 9128d852cf..d7b2a6eff4 100644
--- a/tools/migration/node.1510016282.change-gridfs-metadata-name-to-filename.js
+++ b/tools/migration/node.1510016282.change-gridfs-metadata-name-to-filename.js
@@ -34,7 +34,7 @@ async function main () {
 		1,
 		async (time) => {
 			console.log(`${time} / ${idop}`)
-			const doc = await db.get('drive_files').find(query, {
+			const doc = await DriveFile.find(query, {
 				limit: dop, skip: time * dop
 			})
 			return Promise.all(doc.map(applyNewChange))
diff --git a/tools/migration/node.1510056272.issue_882.js b/tools/migration/node.1510056272.issue_882.js
index aa11413255..302ef3de65 100644
--- a/tools/migration/node.1510056272.issue_882.js
+++ b/tools/migration/node.1510056272.issue_882.js
@@ -31,7 +31,7 @@ async function main() {
 		1,
 		async (time) => {
 			console.log(`${time} / ${idop}`)
-			const doc = await db.get('drive_files').find(query, {
+			const doc = await DriveFile.find(query, {
 				limit: dop, skip: time * dop
 			})
 			return Promise.all(doc.map(migrate))