diff --git a/notebook/tests/services/kernel.js b/notebook/tests/services/kernel.js
index 4f22b802e9..a7e5382632 100644
--- a/notebook/tests/services/kernel.js
+++ b/notebook/tests/services/kernel.js
@@ -310,7 +310,7 @@ casper.notebook_test(function () {
             this.thenEvaluate(function () {
                 var cell = IPython.notebook.get_cell(0);
                 cell.set_text("import os\n" +
-                              "from IPython.kernel.connect import get_connection_file\n" +
+                              "from ipykernel.connect import get_connection_file\n" +
                               "with open(get_connection_file(), 'w') as f:\n" +
                               "    f.write('garbage')\n" +
                               "os._exit(1)");
diff --git a/notebook/tests/services/serialize.js b/notebook/tests/services/serialize.js
index e44ea47043..36bb165659 100644
--- a/notebook/tests/services/serialize.js
+++ b/notebook/tests/services/serialize.js
@@ -33,7 +33,7 @@ casper.notebook_test(function () {
     this.then(function () {
         var index = this.append_cell([
             "import os",
-            "from IPython.kernel.comm import Comm",
+            "from ipykernel.comm.comm import Comm",
             "comm = Comm(target_name='echo')",
             "msgs = []",
             "def on_msg(msg):",
diff --git a/notebook/tests/services/session.js b/notebook/tests/services/session.js
index 5f41278de0..ff1fa25d35 100644
--- a/notebook/tests/services/session.js
+++ b/notebook/tests/services/session.js
@@ -154,7 +154,7 @@ casper.notebook_test(function () {
             this.thenEvaluate(function () {
                 var cell = IPython.notebook.get_cell(0);
                 cell.set_text("import os\n" +
-                              "from IPython.kernel.connect import get_connection_file\n" +
+                              "from ipykernel.connect import get_connection_file\n" +
                               "with open(get_connection_file(), 'w') as f:\n" +
                               "    f.write('garbage')\n" +
                               "os._exit(1)");