From 260f237be2dffc83bbaf6650ea8d3a95460c218f Mon Sep 17 00:00:00 2001
From: Eddie Antonio Santos <easantos@ualberta.ca>
Date: Mon, 1 Oct 2018 15:30:01 -0600
Subject: [PATCH] Not sure why there's still stuff in the event loop, but
 whatever.

---
 index.js | 9 +++------
 test.js  | 1 +
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/index.js b/index.js
index 9f73d559ad..a07da4f2ba 100644
--- a/index.js
+++ b/index.js
@@ -14,10 +14,11 @@ exports.LMLayer = class LMLayer {
   predictWithContext({transform, context, _token}) {
     let token = this._currentToken++;
 
-    let promise = new Promise((resolve, reject) => {
+    return new Promise((resolve, reject) => {
       if (this._promises.has(token)) {
         reject(`Existing request with token ${token}`);
       }
+      this._promises.set(token, resolve);
 
       this._worker.postMessage({
         kind: 'predict',
@@ -25,11 +26,7 @@ exports.LMLayer = class LMLayer {
         transform,
         context
       });
-
-      this._promises.set(token, resolve);
     });
-
-    return promise;
   }
 
   /**
@@ -45,8 +42,8 @@ exports.LMLayer = class LMLayer {
     }
 
     if (kind === 'suggestions') {
-      accept(event.data);
       this._promises.delete(token);
+      accept(event.data);
     } else {
       throw new Error(`Unknown message kind: ${kind}`);
     }
diff --git a/test.js b/test.js
index c051c0096e..cd48fc4391 100644
--- a/test.js
+++ b/test.js
@@ -10,6 +10,7 @@ test('Can provide context to LMLayer', async t => {
     transform: { insert: 'D', deleteLeft: 0, deleteRight: 0 },
     context: { left: '' }
   });
+
   let {suggestions} = message;
 
   t.deepEqual(suggestions, [