Fix IPython object references

Ref jupyter#141 (c) Copyright IBM Corp. 2016
jhpedemonte · Mar 7, 2016 · 0a9216c · 0a9216c
1 parent fb2a665
commit 0a9216c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/elements/urth-core-behaviors/execution-complete-behavior.html b/elements/urth-core-behaviors/execution-complete-behavior.html
@@ -16,15 +16,15 @@
       console.debug("Registering _onExecutionComplete for Execute Reply messages.");
       this.__replyCallback = this._onExecutionComplete.bind(this);
 
-      IPython.notebook.kernel.widget_manager.notebook.events.on(
+      IPython.notebook.events.on(
         'shell_reply.Kernel', this.__replyCallback
       );
     },
 
     detached: function() {
         console.debug("Unregistering _onExecutionComplete for Execute Reply messages.");
         if (this.__replyCallback){
-            IPython.notebook.kernel.widget_manager.notebook.events.off(
+            IPython.notebook.events.off(
                 'shell_reply.Kernel', this.__replyCallback
             );
         }

diff --git a/elements/urth-core-behaviors/jupyter-kernel-observer.html b/elements/urth-core-behaviors/jupyter-kernel-observer.html
@@ -30,7 +30,7 @@
                     'kernel_ready.Kernel', this.__kernelReadyCallback
                 );
 
-                if( IPython.notebook.kernel.is_connected() ){
+                if( IPython.notebook.kernel && IPython.notebook.kernel.is_connected() ){
                   this.onKernelReady();
                 }
             },