diff --git a/ssa/di.go b/ssa/di.go
index 702f776e..d5f5a35f 100644
--- a/ssa/di.go
+++ b/ssa/di.go
@@ -18,7 +18,8 @@ type aDIBuilder struct {
 
 type diBuilder = *aDIBuilder
 
-func newDIBuilder(prog Program, m llvm.Module) diBuilder {
+func newDIBuilder(prog Program, pkg Package) diBuilder {
+	m := pkg.mod
 	ctx := m.Context()
 	m.AddNamedMetadataOperand("llvm.module.flags",
 		ctx.MDNode([]llvm.Metadata{
diff --git a/ssa/package.go b/ssa/package.go
index 552a6104..a91b3b1d 100644
--- a/ssa/package.go
+++ b/ssa/package.go
@@ -719,7 +719,7 @@ func (p Package) diBuilder() diBuilder {
 }
 
 func (p Package) EnableDebugSymbols(name, pkgPath string) {
-	p.di = newDIBuilder(p.Prog, p.mod)
+	p.di = newDIBuilder(p.Prog, p)
 	p.cu = p.di.createCompileUnit(name, pkgPath)
 }