diff --git a/BUCK b/BUCK
index 3aed93a..081ad6f 100644
--- a/BUCK
+++ b/BUCK
@@ -21,6 +21,7 @@
   name = 'cookbook_tests',
   srcs = glob(['src/test/java/**/*.java']),
   labels = ['cookbook-plugin'],
+  source_under_test = [':cookbook-plugin__plugin'],
   deps = [
     ':cookbook-plugin__plugin',
     '//lib:junit',