Fix crash

Fusion-Flux · Jun 13, 2023 · b56fa39 · b56fa39
1 parent da294d4
commit b56fa39
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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);
     }
 
 }