diff --git a/lib/mix/tasks/ash.codegen.ex b/lib/mix/tasks/ash.codegen.ex
index e00ccebe..854f769a 100644
--- a/lib/mix/tasks/ash.codegen.ex
+++ b/lib/mix/tasks/ash.codegen.ex
@@ -7,6 +7,7 @@ defmodule Mix.Tasks.Ash.Codegen do
   @shortdoc "Runs all codegen tasks for any extension on any resource/domain in your application."
   @doc @shortdoc
   def run(argv) do
+    Mix.Task.run("compile")
     {name, argv} =
       case argv do
         ["-" <> _ | _] ->
diff --git a/lib/mix/tasks/generate_livebook.ex b/lib/mix/tasks/ash.generate_livebook.ex
similarity index 100%
rename from lib/mix/tasks/generate_livebook.ex
rename to lib/mix/tasks/ash.generate_livebook.ex
diff --git a/lib/mix/tasks/generate_policy_chart.ex b/lib/mix/tasks/ash.generate_policy_chart.ex
similarity index 100%
rename from lib/mix/tasks/generate_policy_chart.ex
rename to lib/mix/tasks/ash.generate_policy_chart.ex
diff --git a/lib/mix/tasks/generate_resource_diagrams.ex b/lib/mix/tasks/ash.generate_resource_diagrams.ex
similarity index 100%
rename from lib/mix/tasks/generate_resource_diagrams.ex
rename to lib/mix/tasks/ash.generate_resource_diagrams.ex
diff --git a/lib/mix/tasks/ash.rollback.ex b/lib/mix/tasks/ash.rollback.ex
index e6602c21..94d51195 100644
--- a/lib/mix/tasks/ash.rollback.ex
+++ b/lib/mix/tasks/ash.rollback.ex
@@ -7,6 +7,8 @@ defmodule Mix.Tasks.Ash.Rollback do
   @shortdoc "Runs all rollback tasks for any extension on any resource/domain in your application."
   @doc @shortdoc
   def run(argv) do
+    Mix.Task.run("compile")
+
     argv
     |> Ash.Mix.Tasks.Helpers.extensions!()
     |> Enum.map(fn extension ->
diff --git a/lib/mix/tasks/helpers.ex b/lib/mix/tasks/helpers.ex
index 0eaf8654..7a7881ac 100644
--- a/lib/mix/tasks/helpers.ex
+++ b/lib/mix/tasks/helpers.ex
@@ -33,6 +33,7 @@ defmodule Ash.Mix.Tasks.Helpers do
     end
   end
 
+
   @doc """
   Get all domains for the current project and ensure they are compiled.
   """