diff --git a/flow/scripts/save_images.tcl b/flow/scripts/save_images.tcl
index 7980c19880..66cde52206 100644
--- a/flow/scripts/save_images.tcl
+++ b/flow/scripts/save_images.tcl
@@ -35,10 +35,11 @@ gui::set_display_controls "Instances/Physical/*" visible false
 save_image -resolution $resolution $::env(REPORTS_DIR)/final_placement.webp
 
 if {[info exist ::env(PWR_NETS_VOLTAGES)]} {
-#  gui::set_display_controls "Heat Maps/IR Drop" visible true
-#  gui::set_heatmap IRDrop Layer $::env(IR_DROP_LAYER)
-#  save_image -resolution $resolution $::env(REPORTS_DIR)/final_ir_drop.webp
-#  gui::set_display_controls "Heat Maps/IR Drop" visible false
+  gui::set_display_controls "Heat Maps/IR Drop" visible true
+  gui::set_heatmap IRDrop Layer $::env(IR_DROP_LAYER)
+  gui::set_heatmap IRDrop ShowLegend 1
+  save_image -resolution $resolution $::env(REPORTS_DIR)/final_ir_drop.webp
+  gui::set_display_controls "Heat Maps/IR Drop" visible false
 }
 
 # The clock view: all clock nets and buffers