From 3c759f05b42acca5b5c557cf68a9ac3e295feaf5 Mon Sep 17 00:00:00 2001
From: Robin Appelman <robin@icewind.nl>
Date: Fri, 22 Apr 2022 22:41:58 +0200
Subject: [PATCH] better enter/preserve

---
 src/entity.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/entity.rs b/src/entity.rs
index bac13ac..dc0e66f 100644
--- a/src/entity.rs
+++ b/src/entity.rs
@@ -112,7 +112,13 @@ impl ActiveEntities {
             ),
         ];
         for entity in self.entities.values_mut() {
-            if !state.instance_baselines[0].contains(entity.entity_index) {
+            if state.instance_baselines[1]
+                .get(entity.entity_index)
+                .filter(|baseline| baseline.server_class == entity.server_class)
+                .is_some()
+            {
+                entity.update_type = UpdateType::Preserve;
+            } else {
                 entity.update_type = UpdateType::Enter;
             }
         }
@@ -150,7 +156,7 @@ impl ActiveEntities {
 
         baselines[0].updated_base_line = true;
         baselines[1].updated_base_line = true;
-        baselines[1].base_line = 1;
+        baselines[0].base_line = 1;
 
         (
             baselines.into_iter(),