From b56fa39d824ecb790dd27ee22c0b3a0e47062549 Mon Sep 17 00:00:00 2001
From: "Josiah (Gaming32) Glosson" <soujournme@gmail.com>
Date: Tue, 13 Jun 2023 00:54:49 -0500
Subject: [PATCH] Fix crash

---
 .../fusionflux/portalcubed/entity/SchrodingerCubeEntity.java  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/fusionflux/portalcubed/entity/SchrodingerCubeEntity.java b/src/main/java/com/fusionflux/portalcubed/entity/SchrodingerCubeEntity.java
index 0bf093da..af2e84ae 100644
--- a/src/main/java/com/fusionflux/portalcubed/entity/SchrodingerCubeEntity.java
+++ b/src/main/java/com/fusionflux/portalcubed/entity/SchrodingerCubeEntity.java
@@ -16,7 +16,7 @@ public SchrodingerCubeEntity(EntityType<? extends PathfinderMob> type, Level wor
 
     @Override
     public SchrodingerCubeEntity getConnection() {
-        return level.getNearestEntity(
+        return Objects.requireNonNullElse(level.getNearestEntity(
             SchrodingerCubeEntity.class,
             TargetingConditions.forNonCombat()
                 .selector(e -> Objects.equals(e.getCustomName(), getCustomName()))
@@ -25,7 +25,7 @@ public SchrodingerCubeEntity getConnection() {
             this,
             getX(), getY(), getZ(),
             AABB.ofSize(position(), 50, 50, 50)
-        );
+        ), this);
     }
 
 }