1 files changed, 4 insertions, 1 deletions
diff --git a/tools/fileDataProvider.cpp b/tools/fileDataProvider.cpp
index 1ff0279..39fae1b 100644
--- a/tools/fileDataProvider.cpp
+++ b/tools/fileDataProvider.cpp
@@ -58,6 +58,9 @@ void RidFileDP::readEntries() {
         d.addEntry(e);
     }
     pos = 0;
+    std::vector<DataEntry*> & dat = d.getData();
+    while (pos<dat.size() && dat[pos]->rank!=-1 && dat[pos]->qid!="-1")
+        ++pos;
     qid = 1;
     read = true;
 }
@@ -95,7 +98,7 @@ int RidFileDP::getDataSet(DataList &out){
     dat[pos]->qid=std::to_string(qid);
     ++qid;
     dat[pos]->rank=qid;
-    while (pos<dat.size() && dat[pos]->rank!=-1)
+    while (pos<dat.size() && dat[pos]->rank!=-1 && dat[pos]->qid=="-1")
         ++pos;
     if (pos==d.getSize())
         eof = true;