diff --git a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/classification/TableOfContentsClassificationService.java b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/classification/TableOfContentsClassificationService.java index b6392c2..28bea50 100644 --- a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/classification/TableOfContentsClassificationService.java +++ b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/classification/TableOfContentsClassificationService.java @@ -360,7 +360,7 @@ public class TableOfContentsClassificationService { NumberWord curr = numbers.get(i); ClassificationPage prevPage = lookup.get(prev).page(); ClassificationPage currPage = lookup.get(curr).page(); - if (prevPage == currPage) { + if (prevPage.equals(currPage)) { result.add(curr); } else if (curr.word().getBBox().getMinY() < currPage.getPageHeight() * 0.33) { result.add(curr);