diff --git a/extension/android/BUCK b/extension/android/BUCK
index c2d6359af75..5d021250e6d 100644
--- a/extension/android/BUCK
+++ b/extension/android/BUCK
@@ -23,8 +23,13 @@ fb_android_library(
 fb_android_library(
     name = "executorch_llama",
     srcs = [
+        "src/main/java/org/pytorch/executorch/DType.java",
+        "src/main/java/org/pytorch/executorch/EValue.java",
         "src/main/java/org/pytorch/executorch/LlamaCallback.java",
         "src/main/java/org/pytorch/executorch/LlamaModule.java",
+        "src/main/java/org/pytorch/executorch/Module.java",
+        "src/main/java/org/pytorch/executorch/NativePeer.java",
+        "src/main/java/org/pytorch/executorch/Tensor.java",
         "src/main/java/org/pytorch/executorch/annotations/Experimental.java",
     ],
     autoglob = False,