diff --git a/src/dataset.ml b/src/dataset.ml
index 0fc75d81bfda59b9fe6d991e9a2b7173b79a1e0f..6ce481846cd19043eea6f79a4265e965fe684d8c 100644
--- a/src/dataset.ml
+++ b/src/dataset.ml
@@ -143,9 +143,7 @@ let interpret_predicate env ~on_model ~on_dataset task =
     in
     let normalization =
       List.filter_map ls_with_normalization ~f:(fun (ls, normalization) ->
-        if String.equal ls.ls_name.id_string dataset.ls_name.id_string
-        then Some normalization
-        else None)
+        if Term.ls_equal ls dataset then Some normalization else None)
     in
     let dataset = on_dataset dataset in
     let model = on_model model in