diff --git a/lib/transformers/prevent_multidimensional_array_aggregates.ex b/lib/transformers/prevent_multidimensional_array_aggregates.ex
index a52ea72..396d633 100644
--- a/lib/transformers/prevent_multidimensional_array_aggregates.ex
+++ b/lib/transformers/prevent_multidimensional_array_aggregates.ex
@@ -35,7 +35,7 @@ defmodule AshPostgres.Transformers.PreventMultidimensionalArrayAggregates do
       end
     end)
 
-    repo = Transformer.get_option(dsl, [:postgres], :repo)
+    repo = AshPostgres.DataLayer.Info.repo(dsl, :mutate)
 
     cond do
       match?({:error, _}, Code.ensure_compiled(repo)) ->