diff --git a/app/models/rule/registry/transaction_resource.rb b/app/models/rule/registry/transaction_resource.rb
index 5990fb74..15aae288 100644
--- a/app/models/rule/registry/transaction_resource.rb
+++ b/app/models/rule/registry/transaction_resource.rb
@@ -19,9 +19,10 @@ class Rule::Registry::TransactionResource < Rule::Registry
       Rule::ActionExecutor::SetTransactionName.new(rule)
     ]
 
-    # TODO PUT BACK
-    enabled_executors << Rule::ActionExecutor::AutoCategorize.new(rule)
-    enabled_executors << Rule::ActionExecutor::AutoDetectMerchants.new(rule)
+    if ai_enabled?
+      enabled_executors << Rule::ActionExecutor::AutoCategorize.new(rule)
+      enabled_executors << Rule::ActionExecutor::AutoDetectMerchants.new(rule)
+    end
 
     enabled_executors
   end