diff --git a/src/arch/i386/include/undi.h b/src/arch/i386/include/undi.h
index 7d662345..67b063b6 100644
--- a/src/arch/i386/include/undi.h
+++ b/src/arch/i386/include/undi.h
@@ -7,6 +7,9 @@
  *
  */
 
+#include <gpxe/device.h>
+#include <pxe_types.h>
+
 /** An UNDI device */
 struct undi_device {
 	/** Generic device */