diff --git a/jupyter_notebook/tests/base/misc.js b/jupyter_notebook/tests/base/misc.js
index 39e513e87..a19397b17 100644
--- a/jupyter_notebook/tests/base/misc.js
+++ b/jupyter_notebook/tests/base/misc.js
@@ -5,7 +5,7 @@
 casper.notebook_test(function () {
     var jsver = this.evaluate(function () {
         var cell = IPython.notebook.get_cell(0);
-        cell.set_text('import IPython; print(IPython.__version__)');
+        cell.set_text('import jupyter_notebook; print(jupyter_notebook.__version__)');
         cell.execute();
         return IPython.version;
     });
@@ -15,7 +15,7 @@ casper.notebook_test(function () {
     // refactor this into  just a get_output(0)
     this.then(function () {
         var result = this.get_output_cell(0);
-        this.test.assertEquals(result.text.trim(), jsver, 'IPython.version in JS matches IPython.');
+        this.test.assertEquals(result.text.trim(), jsver, 'IPython.version in JS matches server-side.');
     });
 
 });