diff --git a/docs/my-website/docs/proxy/guardrails/guardrail_pipeline_flow_builder.md b/docs/my-website/docs/proxy/guardrails/guardrail_pipeline_flow_builder.md
index 0cd458b24fa..d98354426d0 100644
--- a/docs/my-website/docs/proxy/guardrails/guardrail_pipeline_flow_builder.md
+++ b/docs/my-website/docs/proxy/guardrails/guardrail_pipeline_flow_builder.md
@@ -1,4 +1,5 @@
-import Image from '@theme/IdealImage';
+import Tabs from '@theme/Tabs';
+import TabItem from '@theme/TabItem';
 import Tabs from '@theme/Tabs';
 import TabItem from '@theme/TabItem';