diff --git a/jf-assembler/src/main.cpp b/jf-assembler/src/main.cpp
index 2d0aacb..53c73aa 100644
--- a/jf-assembler/src/main.cpp
+++ b/jf-assembler/src/main.cpp
@@ -11,7 +11,12 @@
 #include "EigerAssembler.hpp"
 #include "assembler_config.hpp"
 #include "ZmqPulseSyncReceiver.hpp"
- 
+
+#ifdef USE_EIGER
+#include "eiger.hpp"
+#else
+#include "jungfrau.hpp"
+#endif
 
 using namespace std;
 using namespace buffer_config;