diff --git a/lib/ash/data_layer/data_layer.ex b/lib/ash/data_layer/data_layer.ex
index 7e435909..c5f528b8 100644
--- a/lib/ash/data_layer/data_layer.ex
+++ b/lib/ash/data_layer/data_layer.ex
@@ -309,7 +309,7 @@ defmodule Ash.DataLayer do
   def source(resource) do
     data_layer = Ash.DataLayer.data_layer(resource)
 
-    Code.ensure_compiled!(resource)
+    Code.ensure_compiled!(data_layer)
 
     if :erlang.function_exported(data_layer, :source, 1) do
       data_layer.source(resource)