diff --git a/nbclient/tests/test_client.py b/nbclient/tests/test_client.py
index bf777802..8be9e1b9 100644
--- a/nbclient/tests/test_client.py
+++ b/nbclient/tests/test_client.py
@@ -28,10 +28,14 @@
 
 addr_pat = re.compile(r'0x[0-9a-f]{7,9}')
 current_dir = os.path.dirname(__file__)
-ipython_input_pat = re.compile(r'<ipython-input-\d+-[0-9a-f]+>')
+ipython_input_pat = re.compile(
+    r'(<ipython-input-\d+-[0-9a-f]+>|<IPY-INPUT>) in (<module>|<cell line: \d>\(\))'
+)
 # Tracebacks look different in IPython 8,
 # see: https://github.com/ipython/ipython/blob/master/docs/source/whatsnew/version8.rst#traceback-improvements  # noqa
-ipython8_input_pat = re.compile(r'Input In \[\d+\],')
+ipython8_input_pat = re.compile(
+    r'(Input In \[\d+\]|<IPY-INPUT>), in (<module>|<cell line: \d>\(\))'
+)
 
 hook_methods = [
     "on_cell_start",