2 files changed, 2 insertions, 2 deletions

diff --git a/test/uk/ac/ox/cs/hermit/HermitQueryReasoner.java b/test/uk/ac/ox/cs/hermit/HermitQueryReasoner.java
index 6e9446d..f6246f8 100644
--- a/test/uk/ac/ox/cs/hermit/HermitQueryReasoner.java
+++ b/test/uk/ac/ox/cs/hermit/HermitQueryReasoner.java
@@ -73,7 +73,7 @@ public class HermitQueryReasoner {
                 if (onto.containsEntityInSignature(date)) {
                         for (OWLOntology o: onto.getImportsClosure())
                                 for (OWLAxiom axiom: o.getAxioms())
-                                        if (axiom.containsEntityInSignature(date)) {
+                                        if (axiom.getDatatypesInSignature().contains(date)) {
                                                 System.out.println("The axiom: " + axiom + " is being ingored. ");
                                                 man.removeAxiom(onto, axiom); 
                                         }
diff --git a/test/uk/ac/ox/cs/hermit/HermitTester.java b/test/uk/ac/ox/cs/hermit/HermitTester.java
index 00df658..dc70284 100644
--- a/test/uk/ac/ox/cs/hermit/HermitTester.java
+++ b/test/uk/ac/ox/cs/hermit/HermitTester.java
@@ -27,7 +27,7 @@ public class HermitTester {
                 for (OWLOntology o: onto.getImportsClosure()) {
                         System.out.println(o.containsEntityInSignature(concept));
                         for (OWLAxiom axiom: o.getAxioms())
-                                if (axiom.containsEntityInSignature(concept))
+                                if (axiom.getClassesInSignature().contains(concept))
                                         System.out.println(axiom); 
                 }