From b2a8224ad33615665014e2aa66e8de910dbb9961 Mon Sep 17 00:00:00 2001
From: EmeraldBlock <yygengjunior@gmail.com>
Date: Wed, 24 Nov 2021 21:44:40 -0600
Subject: [PATCH] only show ejector belt if connected

---
 src/js/game/systems/ejector_belt.js | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/js/game/systems/ejector_belt.js b/src/js/game/systems/ejector_belt.js
index 9d00437f..f1ee4114 100644
--- a/src/js/game/systems/ejector_belt.js
+++ b/src/js/game/systems/ejector_belt.js
@@ -44,13 +44,18 @@ export class EjectorBeltSystem extends GameSystemWithFilter {
             const staticComp = entity.components.StaticMapEntity;
             for (let i = 0; i < ejectorComp.slots.length; ++i) {
                 // Extract underlay parameters
-                const { pos, direction, beltLength } = ejectorComp.slots[i];
+                const { pos, direction, beltLength, cachedTargetEntity } = ejectorComp.slots[i];
 
                 // skips both missing and 0 belt lengths
                 if (!beltLength) {
                     continue;
                 }
 
+                // check if connected
+                if (!cachedTargetEntity) {
+                    continue;
+                }
+
                 const transformedPos = staticComp.localTileToWorld(pos);
                 const destX = transformedPos.x * globalConfig.tileSize;
                 const destY = transformedPos.y * globalConfig.tileSize;