diff --git a/MANIFEST.in b/MANIFEST.in
index 103fd31..bd8858c 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -1 +1 @@
-recursive-include mlflow_cratedb *.sql
+recursive-include mlflow_cratedb *.py *.sql