diff --git a/tests/entity.rs b/tests/entity.rs
index fd8fcbb..5bdfb0c 100644
--- a/tests/entity.rs
+++ b/tests/entity.rs
@@ -94,13 +94,19 @@ impl MessageHandler for EntityDumper {
         }
     }
 
-    fn handle_message(&mut self, message: &Message, tick: u32, _parser_state: &ParserState) {
+    fn handle_message(
+        &mut self,
+        message: &Message,
+        server_tick: u32,
+        _client_tick: u32,
+        _parser_state: &ParserState
+    ) {
         match message {
             Message::PacketEntities(entity_message) => self.entities.extend(
                 entity_message
                     .entities
                     .iter()
-                    .map(|entity| (tick, entity.clone())),
+                    .map(|entity| (server_tick, entity.clone())),
             ),
             _ => {}
         }
diff --git a/tests/message_reencode.rs b/tests/message_reencode.rs
index aa4b4e0..c4eee3e 100644
--- a/tests/message_reencode.rs
+++ b/tests/message_reencode.rs
@@ -73,5 +73,5 @@ fn setup_message(handler: &mut DemoHandler<NullHandler>, input: &str) {
         &handler.state_handler,
     )
     .unwrap();
-    handler.handle_message(message);
+    handler.handle_message(message, 0);
 }