diff --git a/lib/ash_hq_web/controllers/dsl_redirect_controller.ex b/lib/ash_hq_web/controllers/dsl_redirect_controller.ex
index 9ceea7e..e885a4e 100644
--- a/lib/ash_hq_web/controllers/dsl_redirect_controller.ex
+++ b/lib/ash_hq_web/controllers/dsl_redirect_controller.ex
@@ -7,7 +7,8 @@ defmodule AshHqWeb.DslRedirectController do
     extension =
       library.latest_library_version.extensions
       |> Enum.find(fn extension ->
-        AshHqWeb.DocRoutes.sanitize_name(extension.name) == extension_name
+        AshHqWeb.DocRoutes.sanitize_name(extension.module) == extension_name ||
+          AshHqWeb.DocRoutes.sanitize_name(extension.name) == extension_name
       end)
       |> Kernel.||(raise Ash.Error.Query.NotFound)