diff --git a/docs/source/features/workspace_analyzer/workspace_analyzer.md b/docs/source/features/workspace_analyzer/workspace_analyzer.md
index 133ee0eb..a27952ea 100644
--- a/docs/source/features/workspace_analyzer/workspace_analyzer.md
+++ b/docs/source/features/workspace_analyzer/workspace_analyzer.md
@@ -22,6 +22,7 @@ from embodichain.lab.sim.utility.workspace_analyzer import (
     WorkspaceAnalyzerConfig,
     AnalysisMode
 )
+from embodichain.lab.sim.utility.workspace_analyzer.configs import VisualizationConfig
 
 # Setup simulation
 sim = SimulationManager(SimulationManagerCfg(headless=False, sim_device="cpu"))