Best practice for after_action?

defmodule A do
  actions do
    update :update do
      change after_action(fn changeset, record, context ->
        record = Ash.load(record, [:b])
        # you can modify it directly
        {:ok, %{record | b: B.update!(record.b, params)}}
      end)
    end
  end
end