diff --git a/addon/ng2/models/webpack-build-mobile.ts b/addon/ng2/models/webpack-build-mobile.ts
index 1480a4eebf37..0eb18933ddcb 100644
--- a/addon/ng2/models/webpack-build-mobile.ts
+++ b/addon/ng2/models/webpack-build-mobile.ts
@@ -21,7 +21,7 @@ export const getWebpackMobileConfigPartial = function (projectRoot: string, sour
   }
 };
 
-export const getWebpackMobileProdConfigPartial = function (projectRoot: string) {
+export const getWebpackMobileProdConfigPartial = function (projectRoot: string, sourceDir: string) {
   return {
     entry: {
       'sw-install': path.resolve(__dirname, '../utilities/sw-install.js')
diff --git a/addon/ng2/tasks/test.ts b/addon/ng2/tasks/test.ts
index fd8885e28b62..4a92a23eb577 100644
--- a/addon/ng2/tasks/test.ts
+++ b/addon/ng2/tasks/test.ts
@@ -34,7 +34,7 @@ module.exports = Task.extend({
       // Single test entry file. Will run the test.ts bundle and track it.
       options.files = [{ pattern: testFile, watched: false }];
       options.preprocessors = { [testFile]: ['webpack','sourcemap'] };
-      options.webpack = webpackTestConfig(projectRoot);
+      options.webpack = webpackTestConfig(projectRoot, this.project.ngConfig.defaults.sourceDir);
       options.webpackMiddleware = {
         noInfo: true, // Hide webpack output because its noisy.
         stats: { // Also prevent chunk and module display output, cleaner look. Only emit errors.