diff --git a/dexlib2/src/main/java/org/jf/dexlib2/builder/MethodImplementationBuilder.java b/dexlib2/src/main/java/org/jf/dexlib2/builder/MethodImplementationBuilder.java
index 4a1e989d..e830b655 100644
--- a/dexlib2/src/main/java/org/jf/dexlib2/builder/MethodImplementationBuilder.java
+++ b/dexlib2/src/main/java/org/jf/dexlib2/builder/MethodImplementationBuilder.java
@@ -34,7 +34,6 @@ package org.jf.dexlib2.builder;
 import org.jf.dexlib2.iface.MethodImplementation;
 import org.jf.dexlib2.iface.reference.StringReference;
 import org.jf.dexlib2.iface.reference.TypeReference;
-import org.jf.dexlib2.writer.builder.BuilderStringReference;
 
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
@@ -143,7 +142,7 @@ public class MethodImplementationBuilder {
         currentLocation.addEpilogue();
     }
 
-    public void addSetSourceFile(@Nullable BuilderStringReference sourceFile) {
+    public void addSetSourceFile(@Nullable StringReference sourceFile) {
         currentLocation.addSetSourceFile(sourceFile);
     }
 
diff --git a/dexlib2/src/main/java/org/jf/dexlib2/builder/MethodLocation.java b/dexlib2/src/main/java/org/jf/dexlib2/builder/MethodLocation.java
index 18daeeaa..f7c0effc 100644
--- a/dexlib2/src/main/java/org/jf/dexlib2/builder/MethodLocation.java
+++ b/dexlib2/src/main/java/org/jf/dexlib2/builder/MethodLocation.java
@@ -36,7 +36,6 @@ import org.jf.dexlib2.builder.debug.*;
 import org.jf.dexlib2.iface.instruction.Instruction;
 import org.jf.dexlib2.iface.reference.StringReference;
 import org.jf.dexlib2.iface.reference.TypeReference;
-import org.jf.dexlib2.writer.builder.BuilderStringReference;
 
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
@@ -207,7 +206,7 @@ public class MethodLocation {
         getDebugItems().add(new BuilderEpilogueBegin());
     }
 
-    public void addSetSourceFile(@Nullable BuilderStringReference sourceFile) {
+    public void addSetSourceFile(@Nullable StringReference sourceFile) {
         getDebugItems().add(new BuilderSetSourceFile(sourceFile));
     }
 }