test: fix tests

Automattic · Mar 10, 2017 · 0540c8a · 0540c8a
1 parent 0c99538
commit 0540c8a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/model.discriminator.test.js b/test/model.discriminator.test.js
@@ -484,7 +484,9 @@ describe('model', function() {
               $push: {
                 events: { kind: 'Clicked', element: '#button' }
               }
-            }).then(() => doc);
+            }).then(function() {
+              return doc;
+            });
           }).
           then(function(doc) {
             return Batch.findOne({ _id: doc._id });