diff --git a/core-buffer/test/test_bitshuffle.cpp b/core-buffer/test/test_bitshuffle.cpp
index a030256..8ca926e 100644
--- a/core-buffer/test/test_bitshuffle.cpp
+++ b/core-buffer/test/test_bitshuffle.cpp
@@ -6,6 +6,12 @@ extern "C" {
     #include "bitshuffle/bitshuffle.h"
 }
 
+#ifdef USE_EIGER
+#include "eiger.hpp"
+#else
+#include "jungfrau.hpp"
+#endif
+
 using namespace std;
 using namespace buffer_config;