diff --git a/bash_kernel/__main__.py b/bash_kernel/__main__.py
index 683580b..0f87c80 100644
--- a/bash_kernel/__main__.py
+++ b/bash_kernel/__main__.py
@@ -1,3 +1,3 @@
-from IPython.kernel.zmq.kernelapp import IPKernelApp
+from ipykernel.kernelapp import IPKernelApp
 from .kernel import BashKernel
 IPKernelApp.launch_instance(kernel_class=BashKernel)
diff --git a/bash_kernel/kernel.py b/bash_kernel/kernel.py
index 0cdcfb4..ca826fc 100644
--- a/bash_kernel/kernel.py
+++ b/bash_kernel/kernel.py
@@ -1,4 +1,4 @@
-from IPython.kernel.zmq.kernelbase import Kernel
+from ipykernel.kernelbase import Kernel
 from pexpect import replwrap, EOF
 
 from subprocess import check_output