diff --git a/js/src/index.js b/js/src/index.js
index d2fe43f1..bad3fb26 100644
--- a/js/src/index.js
+++ b/js/src/index.js
@@ -1,7 +1,5 @@
-var MPLCanvasModel = require('./mpl_widget').MPLCanvasModel;
-var MPLCanvasView = require('./mpl_widget').MPLCanvasView;
 
-module.exports = {
-    MPLCanvasModel: MPLCanvasModel,
-    MPLCanvasView: MPLCanvasView
-}
+// Export widget models and views, and the npm package version number.
+module.exports = require('./mpl_widget.js');
+module.exports['version'] = require('../package.json').version;
+
diff --git a/js/src/nb_index.js b/js/src/nb_index.js
index 6c0b1ea3..1001dd44 100644
--- a/js/src/nb_index.js
+++ b/js/src/nb_index.js
@@ -8,5 +8,5 @@
 __webpack_public_path__ = document.querySelector('body').getAttribute('data-base-url') + 'nbextensions/jupyter-matplotlib/';
 
 // Export widget models and views, and the npm package version number.
-module.exports = require('./mpl_widget.js');
-module.exports['version'] = require('../package.json').version;
+module.exports = require('./index.js');
+