diff --git a/setup.py b/setup.py
index b5679aa5603f..5cc9ce6c5e39 100644
--- a/setup.py
+++ b/setup.py
@@ -118,6 +118,8 @@
         "chia.wallet",
         "chia.wallet.db_wallet",
         "chia.wallet.puzzles",
+        "chia.wallet.puzzles.clawback",
+        "chia.wallet.puzzles.prefarm",
         "chia.wallet.cat_wallet",
         "chia.wallet.did_wallet",
         "chia.wallet.nft_wallet",