diff --git a/setup.py b/setup.py
index 65c2c3f..6e4373f 100644
--- a/setup.py
+++ b/setup.py
@@ -36,7 +36,9 @@ setup(name="magic-wormhole",
       package_dir={"": "src"},
       packages=["wormhole",
                 "wormhole.cli",
+                "wormhole._dilation",
                 "wormhole.test",
+                "wormhole.test.dilate",
                 ],
       entry_points={
           "console_scripts":