diff --git a/src/ethosu_device_u55_u65.c b/src/ethosu_device_u55_u65.c
index f16f9f8..54d2a79 100644
--- a/src/ethosu_device_u55_u65.c
+++ b/src/ethosu_device_u55_u65.c
@@ -18,10 +18,15 @@
 
 #include "ethosu_interface.h"
 
-#include "ethosu_config.h"
 #include "ethosu_device.h"
 #include "ethosu_log.h"
 
+#ifdef ETHOSU55
+#include "ethosu_config_u55.h"
+#else
+#include "ethosu_config_u65.h"
+#endif
+
 #include <assert.h>
 #include <inttypes.h>
 #include <stdbool.h>