diff --git a/tests/integration/_tmux_session.py b/tests/integration/_tmux_session.py
index 0b59af959f..07c941c97b 100644
--- a/tests/integration/_tmux_session.py
+++ b/tests/integration/_tmux_session.py
@@ -15,6 +15,7 @@
 from typing import TypedDict
 
 import libtmux
+import libtmux.exc
 import pytest
 
 from ._common import generate_test_log_dir
@@ -70,7 +71,7 @@ def __init__(
         self._pane_height = pane_height
         self._pane_width = pane_width
         self._pull_policy = pull_policy
-        self._session: libtmux.Session
+        self._session: libtmux.session.Session
         self._session_name = str(uuid.uuid4())
         self._setup_capture: str | list[str]
         self._setup_commands = setup_commands or []
@@ -114,7 +115,7 @@ def __enter__(self) -> TmuxSession:
         # pylint: disable=attribute-defined-outside-init
         # pylint: disable=too-many-locals
 
-        self._server = libtmux.Server()
+        self._server = libtmux.server.Server()
         self._build_tmux_session()
         self._window = self._session.new_window(self._session_name)
         self._pane = self._window.panes[0]