diff --git a/src/plugin.ts b/src/plugin.ts
index 236cc594..56103bcd 100644
--- a/src/plugin.ts
+++ b/src/plugin.ts
@@ -10,8 +10,6 @@ import { Widget } from '@phosphor/widgets';
 
 import { IJupyterWidgetRegistry } from '@jupyter-widgets/base';
 
-import * as widgetExports from './widget';
-
 import { MODULE_NAME, MODULE_VERSION } from './version';
 
 const EXTENSION_ID = 'jupyter-cytoscape:plugin';
@@ -38,6 +36,6 @@ function activateWidgetExtension(
   registry.registerWidget({
     name: MODULE_NAME,
     version: MODULE_VERSION,
-    exports: widgetExports,
+    exports: () => import('./index'),
   });
 }