diff --git a/t/foobar/expected.ekore b/t/foobar/expected.ekore
index 8d00b54..0975bc5 100644
--- a/t/foobar/expected.ekore
+++ b/t/foobar/expected.ekore
@@ -18,7 +18,7 @@ endmodule [  ]
 
 module C-INJ
   axiom { S1 , S2 , S3 , R } \equals { S3 , R } ( inj { S2 , S3 } ( inj { S1 , S2 } ( T : S1 ) ) , inj { S1 , S3 } ( T : S1 ) ) [  ]
-  symbol inj { From , To } ( From ) : To [  ]
+  symbol inj { From , To } ( From ) : To [ sortInjection{}() ]
 endmodule [  ]
 
 module D-K
diff --git a/t/foobar/foobar.ekore b/t/foobar/foobar.ekore
index 96ab0b9..3964139 100644
--- a/t/foobar/foobar.ekore
+++ b/t/foobar/foobar.ekore
@@ -25,7 +25,7 @@ module B-KSEQ
 endmodule [ ]
 
 module C-INJ
-  symbol inj{From,To}(From) : To []
+  symbol inj{From,To}(From) : To [ sortInjection{}() ]
 
   axiom{S1,S2,S3,R}
     \equals{S3,R}(
diff --git a/t/imp-simple/expected.ekore b/t/imp-simple/expected.ekore
index 5874687..a04d5c1 100644
--- a/t/imp-simple/expected.ekore
+++ b/t/imp-simple/expected.ekore
@@ -23,7 +23,7 @@ endmodule [  ]
 module C-INJ
 
   axiom { S1 , S2 , S3 , R } \equals { S3 , R } ( inj { S2 , S3 } ( inj { S1 , S2 } ( T : S1 ) ) , inj { S1 , S3 } ( T : S1 ) ) [  ]
-  symbol inj { From , To } ( From ) : To [  ]
+  symbol inj { From , To } ( From ) : To [ sortInjection{}() ]
 
 endmodule [  ]
 
diff --git a/t/imp-simple/imp-simple.ekore b/t/imp-simple/imp-simple.ekore
index e02b262..5374b58 100644
--- a/t/imp-simple/imp-simple.ekore
+++ b/t/imp-simple/imp-simple.ekore
@@ -25,7 +25,7 @@ module B-KSEQ
 endmodule [ ]
 
 module C-INJ
-  symbol inj{From,To}(From) : To []
+  symbol inj{From,To}(From) : To [ sortInjection{}() ]
 
   axiom{S1,S2,S3,R}
     \equals{S3,R}(
diff --git a/t/peano/expected.ekore b/t/peano/expected.ekore
index 1c55d6c..b635bdc 100644
--- a/t/peano/expected.ekore
+++ b/t/peano/expected.ekore
@@ -18,7 +18,7 @@ endmodule [  ]
 
 module C-INJ
   axiom { S1 , S2 , S3 , R } \equals { S3 , R } ( inj { S2 , S3 } ( inj { S1 , S2 } ( T : S1 ) ) , inj { S1 , S3 } ( T : S1 ) ) [  ]
-  symbol inj { From , To } ( From ) : To [  ]
+  symbol inj { From , To } ( From ) : To [ sortInjection{}() ]
 endmodule [  ]
 
 module D-K
diff --git a/t/peano/peano.ekore b/t/peano/peano.ekore
index e729e64..5bb32a1 100644
--- a/t/peano/peano.ekore
+++ b/t/peano/peano.ekore
@@ -25,7 +25,7 @@ module B-KSEQ
 endmodule [ ]
 
 module C-INJ
-  symbol inj{From,To}(From) : To []
+  symbol inj{From,To}(From) : To [ sortInjection{}() ]
   axiom{S1,S2,S3,R} \equals{S3,R}( inj{S2,S3}(inj{S1,S2}(T:S1)), inj{S1,S3}(T:S1)) []
 endmodule [ ]