diff --git a/exir/program/_program.py b/exir/program/_program.py
index c5100394aa3..fa46fc7080d 100644
--- a/exir/program/_program.py
+++ b/exir/program/_program.py
@@ -149,9 +149,7 @@ def get_shape_env(gm):
     if shape_env is None:
         return {}
     range_constraints = {
-        k: v
-        for k, v in shape_env.var_to_range.items()
-        if k not in shape_env.replacements
+        shape_env.replacements.get(k, k): v for k, v in shape_env.var_to_range.items()
     }
     # Only when we have an unbacked symint, and it's used as constructor inputs,
     # runtime_var_to_range will make a difference compated to var_to_range.