diff --git a/module_usb_audio/reboot.xc b/module_usb_audio/reboot.xc
index 5444b0ec..56eb9af0 100644
--- a/module_usb_audio/reboot.xc
+++ b/module_usb_audio/reboot.xc
@@ -1,7 +1,8 @@
 #include <xs1.h>
 #include <platform.h>
 #include <xs1_su.h>
-#include <print.h>
+
+#include "xud.h"
 
 #define XS1_SU_PERIPH_USB_ID 0x1
 
@@ -12,7 +13,7 @@ extern tileref tile[];
 
 void device_reboot_aux(void)
 {
-#if (XUD_SERIES_SUPPORT == 1)
+#if (XUD_SERIES_SUPPORT == XUD_U_SERIES)
     /* Disconnect from bus */
     unsigned data[] = {4};
     write_periph_32(usb_tile, XS1_SU_PERIPH_USB_ID, XS1_SU_PER_UIFM_FUNC_CONTROL_NUM, 1, data);
@@ -52,7 +53,7 @@ void device_reboot_aux(void)
 /* Reboots XMOS device by writing to the PLL config register */
 void device_reboot_implementation(chanend spare)
 {
-#if (XUD_SERIES_SUPPORT != 1)
+#if (XUD_SERIES_SUPPORT != XUD_U_SERIES)
     outct(spare, XS1_CT_END);   // have to do this before freeing the chanend
     inct(spare);                // Receive end ct from usb_buffer to close down in both directions