diff --git a/test/counterfactuals/generate_counterfactual.jl b/test/counterfactuals/generate_counterfactual.jl
index 9b383d499..33824dd75 100644
--- a/test/counterfactuals/generate_counterfactual.jl
+++ b/test/counterfactuals/generate_counterfactual.jl
@@ -38,12 +38,7 @@ using CounterfactualExplanations.Objectives: distance_mad
         flat_ce = flatten(ce)
         @test flat_ce isa FlattenedCE
         target_encoded(flat_ce)
-        _ce = unflatten(
-            flat_ce,
-            ce.data,
-            ce.M,
-            ce.generator
-        )
+        _ce = unflatten(flat_ce, ce.data, ce.M, ce.generator)
         @test _ce isa CounterfactualExplanation
     end
 end