diff --git a/test/test_dht.cpp b/test/test_dht.cpp
index 5e600e1fa..72b80dbaa 100644
--- a/test/test_dht.cpp
+++ b/test/test_dht.cpp
@@ -42,6 +42,7 @@ POSSIBILITY OF SUCH DAMAGE.
 #include <iostream>
 
 #include "test.hpp"
+#include "setup_transfer.hpp"
 
 using namespace libtorrent;
 using namespace libtorrent::dht;
diff --git a/test/test_primitives.cpp b/test/test_primitives.cpp
index 688118552..56dbffb8b 100644
--- a/test/test_primitives.cpp
+++ b/test/test_primitives.cpp
@@ -63,6 +63,7 @@ POSSIBILITY OF SUCH DAMAGE.
 #include <iostream>
 
 #include "test.hpp"
+#include "setup_transfer.hpp"
 
 using namespace libtorrent;
 using namespace boost::tuples;