diff --git a/src/modules/blocks/enhanced-code/EnhancedRenderCode.tsx b/src/modules/blocks/enhanced-code/EnhancedRenderCode.tsx
index 2949ed4da..3edc5c8d5 100644
--- a/src/modules/blocks/enhanced-code/EnhancedRenderCode.tsx
+++ b/src/modules/blocks/enhanced-code/EnhancedRenderCode.tsx
@@ -117,7 +117,7 @@ export function EnhancedRenderCode(props: {
       {/*<div>Text Buffer Id</div>*/}
       {/*<div>{fragmentId}</div>*/}
     </Box>
-  ), [props.code, props.semiStableId, props.title]);
+  ), [props.code, props.isPartial, props.semiStableId, props.title]);
 
   const headerRow = React.useMemo(() => {
     const Icon = CodeIcon;