diff --git a/d2ir/compile.go b/d2ir/compile.go
index 24e2e3f60..ed18a6253 100644
--- a/d2ir/compile.go
+++ b/d2ir/compile.go
@@ -465,6 +465,7 @@ func (c *compiler) compileMap(dst *Map, ast, scopeAST *d2ast.Map) {
 					Context_: &RefContext{
 						Scope:    ast,
 						ScopeMap: dst,
+						ScopeAST: scopeAST,
 					},
 				}},
 			}