diff --git a/features/steps/tui_app_coverage_steps.py b/features/steps/tui_app_coverage_steps.py
index 8720b827b..454119880 100644
--- a/features/steps/tui_app_coverage_steps.py
+++ b/features/steps/tui_app_coverage_steps.py
@@ -126,6 +126,16 @@ def _build_mock_textual():
         def __exit__(self, *args):
             pass
 
+    mock_textual_app.App = MockApp
+    mock_textual_app.Screen = MockScreen
+    mock_textual_containers.Vertical = MockVertical
+    mock_textual_containers.Horizontal = MockHorizontal
+    mock_textual_widgets.Header = MockHeader
+    mock_textual_widgets.Footer = MockFooter
+    mock_textual_widgets.Static = MockStatic
+    mock_textual_widgets.TextArea = MockTextArea
+    mock_textual_widgets.Collapsible = MockCollapsible
+
     return {
         "textual": mock_textual,
         "textual.app": mock_textual_app,