diff --git a/lib/api/archiveHandler.ts b/lib/api/archiveHandler.ts
index aeec9cd..46df426 100644
--- a/lib/api/archiveHandler.ts
+++ b/lib/api/archiveHandler.ts
@@ -171,7 +171,7 @@ export default async function archiveHandler(link: LinksAndCollectionAndOwner) {
         await page.goBack();
       } else if (!link.preview?.startsWith("archive")) {
         console.log("No og:image found");
-        page
+        await page
           .screenshot({ type: "jpeg", quality: 20 })
           .then((screenshot) => {
             return createFile({
diff --git a/scripts/migration/migrateToV2.js b/scripts/migration/migrateToV2.js
index 3f4df26..29223e9 100644
--- a/scripts/migration/migrateToV2.js
+++ b/scripts/migration/migrateToV2.js
@@ -88,7 +88,7 @@ async function migrateToV2() {
     if (res) {
       await prisma.link.update({
         where: { id: link.id },
-        data: { pdfPath: path },
+        data: { pdf: path },
       });
       console.log(`${link.id}`);
     } else {
@@ -105,7 +105,7 @@ async function migrateToV2() {
     if (res) {
       await prisma.link.update({
         where: { id: link.id },
-        data: { screenshotPath: path },
+        data: { image: path },
       });
       console.log(`${link.id}`);
     } else {