diff --git a/lazy-ngmodule-hot-loader.js b/lazy-ngmodule-hot-loader.js
index 1f168c6a..828afc98 100644
--- a/lazy-ngmodule-hot-loader.js
+++ b/lazy-ngmodule-hot-loader.js
@@ -10,8 +10,10 @@ const isLazyLoadedNgModule = resource => {
     return issuerContext && issuerContext.endsWith(LAZY_RESOURCE_CONTEXT);
 };
 
-module.exports = function (source) {
-    return isLazyLoadedNgModule(this._module) ?
+module.exports = function (source, map) {
+    const modifiedSource = isLazyLoadedNgModule(this._module) ?
         `${source};${HOT_SELF_ACCEPT}`:
         source;
+
+    this.callback(null, modifiedSource, map);
 };