From bbc76b1b73148cac7598894075ff4877dd8097c5 Mon Sep 17 00:00:00 2001
From: Berkeley Martinez <Berkeley@robotie.com>
Date: Thu, 7 Apr 2016 16:38:39 -0700
Subject: [PATCH] Fix regression in challenge map migration

---
 .../migrate-completed-challenges.js            | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/server/middlewares/migrate-completed-challenges.js b/server/middlewares/migrate-completed-challenges.js
index a5f7110f86..9085b98f23 100644
--- a/server/middlewares/migrate-completed-challenges.js
+++ b/server/middlewares/migrate-completed-challenges.js
@@ -104,11 +104,19 @@ export default function migrateCompletedChallenges() {
     if (!user || user.isChallengeMapMigrated) {
       return next();
     }
-    return buildChallengeMap(
-      user.id.toString(),
-      user.completedChallenges,
-      User
-    )
+    const id = user.id.toString();
+    return User.findOne$({
+      where: { id },
+      fields: { completedChallenges: true }
+    })
+      .map(({ completedChallenges = [] } = {}) => completedChallenges)
+      .flatMap(completedChallenges => {
+        return buildChallengeMap(
+          id,
+          completedChallenges,
+          User
+        );
+      })
       .subscribe(
         count => log('documents update', count),
         // errors go here