diff --git a/src/html-import-template-loader.js b/src/html-import-template-loader.js
index <HASH>..<HASH> 100644
--- a/src/html-import-template-loader.js
+++ b/src/html-import-template-loader.js
@@ -44,10 +44,10 @@ export class HTMLImportTemplateLoader {
 
     if (bundleLink) {
       this.onBundleReady = this._importBundle(bundleLink).then(doc => {
-        return this._normalizeTemplateIds(loader, doc);
-      }).then(() => {
-        this.bundle = doc;
-        this.onBundleReady = null;
+        return this._normalizeTemplateIds(loader, doc).then(() => {
+          this.bundle = doc;
+          this.onBundleReady = null;
+        });
       });
 
       return this.onBundleReady.then(() => this._tryGetTemplateFromBundle(entry));
fix(loader): missing context variable

Fixes #7