diff --git a/setup.py b/setup.py
index 391241f01..55ee4bc96 100644
--- a/setup.py
+++ b/setup.py
@@ -103,6 +103,7 @@ def gen_modules(version):
                    "src/xorfinder.cpp",
                    "src/oracle/oracle.cpp",
                    "src/oracle_use.cpp",
+                   "src/probe.cpp",
                ],
         depends = [
             "python/cadiback/cadiback.h",