diff --git a/test_engine.c b/test_engine.c
index d4aaf18..51517f9 100644
--- a/test_engine.c
+++ b/test_engine.c
@@ -3,6 +3,8 @@
 #include <string.h>
 #include <assert.h>
 #include <stdatomic.h>
+#include "dispatcher.h"
+#include "transport.h"
 #include "engine.h"
 
 // Test helper