Merge pull request #31 from jonathaningram/fix-example

Fix broken example
JalfResi · Aug 29, 2017 · c0282de · c0282de
2 parents 363c8d4 + 98e6348
commit c0282de
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/example/justext/main.go b/example/justext/main.go
@@ -2,9 +2,10 @@ package main
 
 import (
 	"flag"
-	"github.com/JalfResi/gojustext"
 	"log"
 	"os"
+
+	"github.com/JalfResi/justext"
 )
 
 var (
@@ -24,12 +25,17 @@ var (
 func main() {
 	flag.Parse()
 
-	var stoplist map[string]bool
+	var (
+		err      error
+		stoplist map[string]bool
+	)
 
 	if *stoplistFilename != "" {
-		stoplist = justext.ReadStoplist(*stoplistFilename)
+		stoplist, err = justext.ReadStoplist(*stoplistFilename)
+		if err != nil {
+			log.Fatal(err)
+		}
 	} else {
-		var err error
 		stoplist, err = justext.GetStoplist(*stoplistLanguage)
 		if err != nil {
 			log.Fatal(err)