diff --git a/geoentitylinker-addon/src/main/java/opennlp/addons/geoentitylinker/GeoEntityLinkerSetupUtils.java b/geoentitylinker-addon/src/main/java/opennlp/addons/geoentitylinker/GeoEntityLinkerSetupUtils.java
index 06581a4..16b2d12 100644
--- a/geoentitylinker-addon/src/main/java/opennlp/addons/geoentitylinker/GeoEntityLinkerSetupUtils.java
+++ b/geoentitylinker-addon/src/main/java/opennlp/addons/geoentitylinker/GeoEntityLinkerSetupUtils.java
@@ -36,6 +36,10 @@
 import opennlp.tools.util.ObjectStream;
 import opennlp.tools.util.PlainTextByLineStream;
 import static opennlp.addons.geoentitylinker.ModelBasedScorer.RADIUS;
+import opennlp.tools.cmdline.MarkableFileInputStreamFactory;
+import opennlp.tools.ml.model.DataIndexer;
+import opennlp.tools.util.InputStreamFactory;
+import opennlp.tools.util.MockInputStreamFactory;
 
 /**
  *
@@ -107,8 +111,8 @@
     InputStream dataIn = new FileInputStream(annotationOutFile);
     try {
 
-      ObjectStream<String> lineStream =
-              new PlainTextByLineStream(dataIn, "UTF-8");
+    
+      ObjectStream<String> lineStream = new PlainTextByLineStream(new MockInputStreamFactory(dataIn), "UTF-8");
       ObjectStream<DocumentSample> sampleStream = new DocumentSampleStream(lineStream);
 
       model = DocumentCategorizerME.train("en", sampleStream);