2 files changed, 4 insertions, 3 deletions

diff --git a/src/main/scala/rsacomb/RSAOntology.scala b/src/main/scala/rsacomb/RSAOntology.scala
index 9220434..52e13e6 100644
--- a/src/main/scala/rsacomb/RSAOntology.scala
+++ b/src/main/scala/rsacomb/RSAOntology.scala
@@ -316,6 +316,7 @@ trait RSAOntology {
     // }
 
     def cycle(axiom: OWLSubClassOfAxiom): Set[Term] = {
+      // TODO: we can actually use `toTriple` from `RSAAxiom`
       val classes =
         axiom.classesInSignature.collect(Collectors.toList()).asScala
       val classA = classes(0)
diff --git a/src/test/scala/rsacomb/CanonicalModelSpec.scala b/src/test/scala/rsacomb/CanonicalModelSpec.scala
index 410619f..ef27a4f 100644
--- a/src/test/scala/rsacomb/CanonicalModelSpec.scala
+++ b/src/test/scala/rsacomb/CanonicalModelSpec.scala
@@ -22,8 +22,8 @@ object Ontology1_CanonicalModelSpec {
   /* Renderer to display OWL Axioms with DL syntax*/
   val renderer = new DLSyntaxObjectRenderer()
 
-  val ontology_path: File = new File("examples/example1.owl")
-  val ontology = RSA.loadOntology(ontology_path)
+  val ontology_path: File = new File("examples/example1.ttl")
+  val ontology: RSAOntology = RSA.loadOntology(ontology_path)
   val program = ontology.canonicalModel
 
   val roleR = new OWLObjectPropertyImpl(RSA.base("R"))
@@ -220,7 +220,7 @@ object Ontology2_CanonicalModelSpec {
   val renderer = new DLSyntaxObjectRenderer()
 
   val ontology_path: File = new File("examples/example2.owl")
-  val ontology = RSA.loadOntology(ontology_path)
+  val ontology: RSAOntology = RSA.loadOntology(ontology_path)
   val program = ontology.canonicalModel
 
   val roleR = new OWLObjectPropertyImpl(RSA.base("R"))