1 files changed, 3 insertions, 3 deletions
diff --git a/sortnew.go b/sortnew.go
index 6479a2d..5b6084b 100644
--- a/sortnew.go
+++ b/sortnew.go
@@ -5,10 +5,10 @@ import (
 	"strings"
 )
 
-func makeNewSortfile() {
-	f, _ := os.OpenFile(sortmap["protobase"]+".newsort.pb.go", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
+func (pb *Files) makeNewSortfile(pf *File) {
+	f, _ := os.OpenFile(pf.Filebase+".newsort.pb.go", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
 
-	header(f, sortmap)
+	header(f, pf)
 
 	if sortmap["lock"] == "all" {
 		// if the lock is set to 'all' this means the mutex was put in the protoc-gen-go struct