diff --git a/CHANGELOG.md b/CHANGELOG.md
index 81c4d29..5dd3ab1 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,7 +1,9 @@
 # CHANGELOG
 
 ## 0.0.32dev
+
 * [Feature] Add benchmarking CLI `python -m ploomber_engine.benchmark`
+* [Feature] Add `ploomber-extension` as a dependency
 
 ## 0.0.31 (2023-08-24)
 
diff --git a/override_failed_result_images.py b/override_failed_result_images.py
index a6b9b1b..87f2292 100644
--- a/override_failed_result_images.py
+++ b/override_failed_result_images.py
@@ -3,6 +3,7 @@
 this happens sometimes when we make changes, we have to check manually
 if differences are acceptable, then run this script
 """
+
 from pathlib import Path
 from glob import glob
 import shutil
diff --git a/setup.py b/setup.py
index 06c9c9d..b49fa60 100644
--- a/setup.py
+++ b/setup.py
@@ -17,6 +17,7 @@
 
 REQUIRES = [
     "ploomber-core>=0.2",
+    "ploomber-extension",
     "debuglater>=1.4.4",
     # used for our debug now feature (PloomberNotebookClient)
     "nbclient",
@@ -37,7 +38,8 @@
 
 DEV = [
     "pkgmt",
-    "pytest",
+    # using pytest 7 because of an issue when using @pytest.mark.memory
+    "pytest<8",
     "flake8",
     "invoke",
     "twine",
diff --git a/src/ploomber_engine/benchmark.py b/src/ploomber_engine/benchmark.py
index deacc21..cf1965c 100644
--- a/src/ploomber_engine/benchmark.py
+++ b/src/ploomber_engine/benchmark.py
@@ -1,6 +1,7 @@
 """
 A module to benchmark notebooks in a directory.
 """
+
 from pathlib import Path
 
 import click
diff --git a/src/ploomber_engine/cli.py b/src/ploomber_engine/cli.py
index 40ff94c..4639427 100644
--- a/src/ploomber_engine/cli.py
+++ b/src/ploomber_engine/cli.py
@@ -1,6 +1,7 @@
 """
 Command-line interface
 """
+
 import ast
 import click
 
diff --git a/src/ploomber_engine/execute.py b/src/ploomber_engine/execute.py
index affe101..b92f268 100644
--- a/src/ploomber_engine/execute.py
+++ b/src/ploomber_engine/execute.py
@@ -1,6 +1,7 @@
 """
 Abstractions for running notebooks with papermill-like interface
 """
+
 import warnings
 from pathlib import Path
 import csv
diff --git a/src/ploomber_engine/ipython.py b/src/ploomber_engine/ipython.py
index 5e88410..75557b3 100644
--- a/src/ploomber_engine/ipython.py
+++ b/src/ploomber_engine/ipython.py
@@ -489,9 +489,11 @@ def execute(self, parameters=None):
         if self._debug_later:
             add_debuglater_cells(
                 self._nb,
-                path_to_dump=self._debug_later
-                if isinstance(self._debug_later, (str, Path))
-                else None,
+                path_to_dump=(
+                    self._debug_later
+                    if isinstance(self._debug_later, (str, Path))
+                    else None
+                ),
             )
 
         with self:
diff --git a/src/ploomber_engine/testing.py b/src/ploomber_engine/testing.py
index 82fd5f8..1b1faa7 100644
--- a/src/ploomber_engine/testing.py
+++ b/src/ploomber_engine/testing.py
@@ -1,6 +1,7 @@
 """
 Testing notebooks against their recorded outputs
 """
+
 import warnings
 
 import nbformat
diff --git a/src/ploomber_engine/tracking/io.py b/src/ploomber_engine/tracking/io.py
index c65a603..6817fa2 100644
--- a/src/ploomber_engine/tracking/io.py
+++ b/src/ploomber_engine/tracking/io.py
@@ -2,6 +2,7 @@
 NOTE: this was copied from
 https://github.com/ploomber/jupyblog/blob/master/src/jupyblog/execute.py
 """
+
 from pathlib import Path
 import base64
 import re
diff --git a/tests/test_cli.py b/tests/test_cli.py
index c4d8888..ce3c66c 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -1,6 +1,7 @@
 """
 Command-line interface tests
 """
+
 from unittest.mock import Mock, call
 
 import pytest
diff --git a/tests/test_tracking.py b/tests/test_tracking.py
index d797806..e09fe55 100644
--- a/tests/test_tracking.py
+++ b/tests/test_tracking.py
@@ -311,4 +311,4 @@ def test_tracking_import_telemetry(tmp_empty, monkeypatch):
         quiet=True,
     )
 
-    assert mock.call_count == 2
+    assert mock.call_count == 1