diff --git a/src/main.cpp b/src/main.cpp
index 18d0cb0..0cde3b6 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -1,7 +1,7 @@
 #include <iostream>
 
-#include <memory>
 #include <examples/PacketExample.h>
+#include <memory>
 
 class KeepAliveHandler : public sp::PacketHandler {
 	void Handle(KeepAlivePacket& packet) {
@@ -16,26 +16,28 @@ class KeepAliveHandler : public sp::PacketHandler {
 int main() {
 	auto keepAlive = std::make_unique<KeepAlivePacket>(69);
 	sp::PacketMessage* msg = keepAlive.get();
-	
+
 	KeepAliveHandler handler;
 	msg->Dispatch(handler);
 
 	sp::DataBuffer buffer;
 	msg->Write(buffer);
 
+	std::uint8_t msgId;
+	buffer >> msgId;
+
 	auto keepAlive2 = std::make_unique<KeepAlivePacket>();
 	keepAlive2->Read(buffer);
 
 	std::cout << "KeepAlive2 : " << keepAlive2->GetField<KeepAliveId>() << "\n";
 
-	//TODO: write ID
 	sp::PacketFactory factory;
-	auto packet = factory.CreateMessage(Disconnect);
+	auto packet = factory.CreateMessage(msgId);
 	if (packet == nullptr) {
 		std::cout << "Mauvais ID !\n";
 		return 1;
 	}
-	std::cout << (unsigned) packet->GetId() << std::endl;
+	std::cout << (unsigned)packet->GetId() << std::endl;
 	packet->Dispatch(handler);
 
 	return 0;