diff --git a/src/protocol/generate_protocol.py b/src/protocol/generate_protocol.py
index 764841f..2315b5d 100644
--- a/src/protocol/generate_protocol.py
+++ b/src/protocol/generate_protocol.py
@@ -476,6 +476,8 @@ def main():
     add_text('#include <cstdint>')
     add_text('#include "Types.h"')
     add_text('#include "PacketReader.h"')
+    add_text('#include "NetworkMessage.h"')
+    add_text('#include <string_view>')
     newline()
     add_text('namespace Feather::Protocol')
     add_text('{{')