From 58a1f99cfa091ad6a6962f7e4b17e817ef2b0247 Mon Sep 17 00:00:00 2001
From: MidnightTale <midnighttale24@gmail.com>
Date: Tue, 25 Apr 2023 00:08:10 +0700
Subject: [PATCH] right

---
 src/main/java/xyz/hynse/foliaflow/FoliaFlow.java | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/main/java/xyz/hynse/foliaflow/FoliaFlow.java b/src/main/java/xyz/hynse/foliaflow/FoliaFlow.java
index 7074242..46f2484 100644
--- a/src/main/java/xyz/hynse/foliaflow/FoliaFlow.java
+++ b/src/main/java/xyz/hynse/foliaflow/FoliaFlow.java
@@ -84,9 +84,12 @@ public class FoliaFlow extends JavaPlugin implements Listener {
         Location loc = new Location(Bukkit.getWorld("world_the_end"), 100, 48, 0);
         BlockData blockData = Material.OBSIDIAN.createBlockData();
         display = Objects.requireNonNull(Bukkit.getWorld("world_the_end")).spawn(loc, BlockDisplay.class);
+        if (display == null) {
+            getLogger().severe("Failed to spawn BlockDisplay entity.");
+            return;
+        }
         display.setBlock(blockData);
-
-
+        getLogger().info("BlockDisplay entity spawned successfully.");
         getServer().getPluginManager().registerEvents(this, this);
     }
 
@@ -99,6 +102,7 @@ public class FoliaFlow extends JavaPlugin implements Listener {
         for (Entity entity : Objects.requireNonNull(Bukkit.getWorld("world_the_end")).getEntitiesByClass(BlockDisplay.class)) {
             entity.remove();
         }
+        getLogger().info("BlockDisplay entity removed successfully.");
         task.cancel();
         blockktask.cancel();
         super.onDisable();