diff --git a/display_api/init.lua b/display_api/init.lua
index 4c0be22..bf95ded 100644
--- a/display_api/init.lua
+++ b/display_api/init.lua
@@ -117,6 +117,10 @@ end
 
 --- Entity update
 function update_entity(entity)
+	if not entity then
+		return
+	end
+	
 	if not entity.nodepos then
 		entity.object:remove() -- Remove old/buggy entity
 		return