diff --git a/nbrunner/nb_runner.py b/nbrunner/nb_runner.py
index 0d025d8..b5f7503 100644
--- a/nbrunner/nb_runner.py
+++ b/nbrunner/nb_runner.py
@@ -7,11 +7,8 @@
 import deepsearch as ds
 import nbformat
 from deepsearch.cps.client.components.elastic import ElasticProjectDataCollectionSource
-from nbconvert.preprocessors import (
-    CellExecutionError,
-    CellTimeoutError,
-    ExecutePreprocessor,
-)
+from nbclient.exceptions import CellTimeoutError
+from nbconvert.preprocessors import CellExecutionError, ExecutePreprocessor
 from rich.console import Console
 from rich.style import Style
 from rich.table import Table
diff --git a/poetry.lock b/poetry.lock
index 2fe17fc..8a333b5 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -2035,13 +2035,13 @@ files = [
 
 [[package]]
 name = "nbclient"
-version = "0.8.0"
+version = "0.9.0"
 description = "A client library for executing notebooks. Formerly nbconvert's ExecutePreprocessor."
 optional = false
 python-versions = ">=3.8.0"
 files = [
-    {file = "nbclient-0.8.0-py3-none-any.whl", hash = "sha256:25e861299e5303a0477568557c4045eccc7a34c17fc08e7959558707b9ebe548"},
-    {file = "nbclient-0.8.0.tar.gz", hash = "sha256:f9b179cd4b2d7bca965f900a2ebf0db4a12ebff2f36a711cb66861e4ae158e55"},
+    {file = "nbclient-0.9.0-py3-none-any.whl", hash = "sha256:a3a1ddfb34d4a9d17fc744d655962714a866639acd30130e9be84191cd97cd15"},
+    {file = "nbclient-0.9.0.tar.gz", hash = "sha256:4b28c207877cf33ef3a9838cdc7a54c5ceff981194a82eac59d558f05487295e"},
 ]
 
 [package.dependencies]
@@ -4498,4 +4498,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p
 [metadata]
 lock-version = "2.0"
 python-versions = ">= 3.8, <3.11"
-content-hash = "5101402246fa7dd36212f2c0c6310b195599505c39ce8a1a0236a1c97624d607"
+content-hash = "cc633922bcec3da0c3db35f099bbbbeceee1e6d28da3ed1e668930f8dbb935fe"
diff --git a/pyproject.toml b/pyproject.toml
index a4b67ce..acf551e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,6 +31,7 @@ spacy = "^3.5.2"
 pydantic = "^2.0.3"
 python-dotenv = "^1.0.0"
 pydantic-settings = "^2.1.0"
+nbclient = "^0.9.0"
 
 [tool.poetry.group.dev.dependencies]
 black = {extras = ["jupyter"], version = "^22.1.0"}
diff --git a/requirements.txt b/requirements.txt
index df69b97..999613a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -78,7 +78,7 @@ mistune==3.0.2 ; python_version >= "3.8" and python_version < "3.11"
 mols2grid==2.0.0 ; python_version >= "3.8" and python_version < "3.11"
 monotonic==1.6 ; python_version >= "3.8" and python_version < "3.11"
 murmurhash==1.0.10 ; python_version >= "3.8" and python_version < "3.11"
-nbclient==0.8.0 ; python_version >= "3.8" and python_version < "3.11"
+nbclient==0.9.0 ; python_version >= "3.8" and python_version < "3.11"
 nbconvert==7.9.2 ; python_version >= "3.8" and python_version < "3.11"
 nbformat==5.9.2 ; python_version >= "3.8" and python_version < "3.11"
 nest-asyncio==1.5.8 ; python_version >= "3.8" and python_version < "3.11"