diff --git a/demo-shell-ng2/systemjs.config.js b/demo-shell-ng2/systemjs.config.js
index 8b717dbca4..df0508421f 100644
--- a/demo-shell-ng2/systemjs.config.js
+++ b/demo-shell-ng2/systemjs.config.js
@@ -12,7 +12,7 @@
 
         'ng2-translate': 'node_modules/ng2-translate',
         'ng2-alfresco-core': 'node_modules/ng2-alfresco-core',
-        'ng2-alfresco-datatable': 'node_modules/ng2-alfresco-datatable',
+        'ng2-alfresco-datatable': 'node_modules/ng2-alfresco-datatable/dist',
         'ng2-alfresco-documentlist': 'node_modules/ng2-alfresco-documentlist',
         'ng2-alfresco-login': 'node_modules/ng2-alfresco-login',
         'ng2-alfresco-search': 'node_modules/ng2-alfresco-search',
@@ -27,7 +27,7 @@
 
         'ng2-translate': { defaultExtension: 'js' },
         'ng2-alfresco-core': {defaultExtension: 'js'},
-        'ng2-alfresco-datatable': {defaultExtension: 'js'},
+        'ng2-alfresco-datatable': { main: 'index.js', defaultExtension: 'js'},
         'ng2-alfresco-documentlist': {defaultExtension: 'js'},
         'ng2-alfresco-login': {defaultExtension: 'js'},
         'ng2-alfresco-search': {defaultExtension: 'js'},