diff --git a/tests/analytical_visualization/app.py b/tests/analytical_visualization/app.py
index 0c1da5d..7ae9a32 100644
--- a/tests/analytical_visualization/app.py
+++ b/tests/analytical_visualization/app.py
@@ -25,6 +25,11 @@ for _mod_name in (
     "tests.analytical_visualization.autosuggest",
     "tests.analytical_visualization.stage_latencies",
     "tests.analytical_visualization.memory_layout",
+    "tests.analytical_visualization.pe_weight_layout",
+    "tests.analytical_visualization.tensor_sharding",
+    "tests.analytical_visualization.topology_map",
+    "tests.analytical_visualization.pipeline_diagram",
+    "tests.analytical_visualization.optimization_report",
 ):
     _m = sys.modules.get(_mod_name)
     if _m is not None: