diff --git a/lib/ash/data_layer/data_layer.ex b/lib/ash/data_layer/data_layer.ex
index 29dbaac6..562f78ee 100644
--- a/lib/ash/data_layer/data_layer.ex
+++ b/lib/ash/data_layer/data_layer.ex
@@ -41,6 +41,7 @@ defmodule Ash.DataLayer do
               metadata: %{step_name: atom | list(term), flow: module(), actor: term()}
             }
           | %{type: :custom, metadata: map()}
+          | %{type: atom, metadata: map()}
 
   @type feature() ::
           :transact