diff --git a/tests/module/PacketGate_1.test b/tests/module/PacketGate_1.test
index 21d50a1ef529..12540509e3a3 100644
--- a/tests/module/PacketGate_1.test
+++ b/tests/module/PacketGate_1.test
@@ -5,6 +5,7 @@ ends the packet streaming in both the source and the sink.
 
 %file: TestSource.cc
 #include "inet/common/ModuleRefByGate.h"
+#include "inet/common/packet/chunk/ByteCountChunk.h"
 #include "inet/queueing/contract/IPacketSource.h"
 #include "inet/queueing/gate/PacketGate.h"
 
diff --git a/tests/module/PriorityScheduler_1.test b/tests/module/PriorityScheduler_1.test
index 6be4b90d6651..44b51845a28b 100644
--- a/tests/module/PriorityScheduler_1.test
+++ b/tests/module/PriorityScheduler_1.test
@@ -6,6 +6,7 @@ the source and the sink.
 
 %file: TestSource.cc
 #include "inet/common/ModuleRefByGate.h"
+#include "inet/common/packet/chunk/ByteCountChunk.h"
 #include "inet/queueing/contract/IPassivePacketSource.h"
 #include "inet/queueing/scheduler/PriorityScheduler.h"