diff --git a/main.c b/main.c
index d37f2f6..22082d9 100644
--- a/main.c
+++ b/main.c
@@ -33,6 +33,7 @@
 #include <hf/call.h>
 #include <hf/ffa.h>
 #include <hf/transport.h>
+#include <hf/vm_ids.h>
 
 #include "uapi/hf/socket.h"