diff --git a/src/test/java/inferencerule/EqualityAxiomTest.java b/src/test/java/inferencerule/EqualityAxiomTest.java
index f01d76a..48f1eaa 100644
--- a/src/test/java/inferencerule/EqualityAxiomTest.java
+++ b/src/test/java/inferencerule/EqualityAxiomTest.java
@@ -10,11 +10,10 @@
 import inference.ProofSystem;
 import models.formulas.DependencyFormula;
 import models.formulas.EquationFormula;
-import models.formulas.Formula;
+import models.terms.DependencyTerm;
 import models.terms.EvaluatableTerm;
 import models.terms.RDLTerm;
 import models.terms.Resource;
-import utils.Utils;
 
 public class EqualityAxiomTest {
 	
@@ -52,21 +51,16 @@
 	
 	@Test
 	void TransitivityTest() {
-		Set<Formula> formulas = ProofSystem.transitivity.apply(Set.of(new EquationFormula(a, b), new EquationFormula(b, c), new EquationFormula(c, d)), Set.of());
-		assertTrue(formulas.contains(new EquationFormula(a, c)));
-		assertTrue(formulas.contains(new EquationFormula(b, d)));
-		assertFalse(formulas.contains(new EquationFormula(a, d)));
+		Set<EvaluatableTerm> formulas = ProofSystem.transitivity.apply(List.of(new EquationFormula(a, b), new EquationFormula(b, c)), c);
+		assertTrue(formulas.contains(a));
 	}
 	
 	@Test
 	void RightSubTest() {
 		EquationFormula eq1 = new EquationFormula(a, b);
 		DependencyFormula d1 = new DependencyFormula(c, d);
-		Set<Formula> formulas = ProofSystem.rightSubstitution.apply(Set.of(eq1, d1), Set.of(e, f));
-		assertTrue(formulas.isEmpty());
-		EquationFormula eq2 = Utils.in(e, d);
-		formulas = ProofSystem.rightSubstitution.apply(Set.of(eq1, d1, eq2), Set.of(e, f));
-		assertEquals(formulas.size(), 1);
+		Set<EvaluatableTerm> formulas = ProofSystem.rightSubstitution.apply(List.of(eq1, d1), new DependencyTerm(c, d, a));
+		assertTrue(formulas.contains(new DependencyTerm(c, d, b)));
 	}
 	
 	@Test