diff --git a/hdk/common/debugging/draw_graph.py b/hdk/common/debugging/draw_graph.py
index 33cf49bd7..8454dc5f4 100644
--- a/hdk/common/debugging/draw_graph.py
+++ b/hdk/common/debugging/draw_graph.py
@@ -13,6 +13,7 @@ IR_NODE_COLOR_MAPPING = {
     ir.Add: "red",
     ir.Sub: "yellow",
     ir.Mul: "green",
+    ir.ArbitraryFunction: "orange",
     "output": "magenta",
 }