From 6ebac90fd9d4df3410793f07267b28b5c2cbbaad Mon Sep 17 00:00:00 2001
From: Ukendio <ukendio@gmail.com>
Date: Sat, 10 May 2025 21:37:53 +0200
Subject: [PATCH] Fix entity lookup

---
 .../systems/receive_replication.luau           | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/demo/src/ReplicatedStorage/systems/receive_replication.luau b/demo/src/ReplicatedStorage/systems/receive_replication.luau
index 7574f34..68376f8 100644
--- a/demo/src/ReplicatedStorage/systems/receive_replication.luau
+++ b/demo/src/ReplicatedStorage/systems/receive_replication.luau
@@ -5,15 +5,19 @@ local collect = require("../collect")
 local client_ids = {}
 
 local function ecs_map_get(world: types.World, id: types.Entity)
-	local deserialised_id = 0
-
-	if not world:exists(id) or not world:contains(id) then
-		deserialised_id = world:entity(id)
+	local deserialised_id = client_ids[id]
+	if not deserialised_id then
+		if world:has(id, jecs.Name) then
+			deserialised_id = world:entity(id)
+		else
+			if world:exists(id) then
+				deserialised_id = world:entity()
+			else
+				deserialised_id = world:entity(id)
+			end
+		end
 		client_ids[id] = deserialised_id
-	else
-		deserialised_id = client_ids[id]
 	end
-
 	return deserialised_id
 end