From 8bc84cbdd28e83f0c7c9d1126e99ec04d39df9b0 Mon Sep 17 00:00:00 2001
From: LoJoSho <LoJoShogaming@gmail.com>
Date: Fri, 24 Feb 2023 09:02:16 -0600
Subject: [PATCH] Fix balloon teleport fix not checking if player has balloon

---
 .../hmccosmetics/listener/PlayerGameListener.java  | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/common/src/main/java/com/hibiscusmc/hmccosmetics/listener/PlayerGameListener.java b/common/src/main/java/com/hibiscusmc/hmccosmetics/listener/PlayerGameListener.java
index ad9a2b8f..6c856e76 100644
--- a/common/src/main/java/com/hibiscusmc/hmccosmetics/listener/PlayerGameListener.java
+++ b/common/src/main/java/com/hibiscusmc/hmccosmetics/listener/PlayerGameListener.java
@@ -133,14 +133,14 @@ public class PlayerGameListener implements Listener {
             return;
         }
 
-        final Cosmetic cosmetic = user.getCosmetic(CosmeticSlot.BALLOON);
-        final Color color = user.getCosmeticColor(CosmeticSlot.BALLOON);
-        user.removeCosmeticSlot(CosmeticSlot.BALLOON);
+        if (user.hasCosmeticInSlot(CosmeticSlot.BALLOON)) {
+            user.despawnBalloon();
 
-        Bukkit.getScheduler().runTaskLater(HMCCosmeticsPlugin.getInstance(), () -> {
-            user.addPlayerCosmetic(cosmetic, color);
-            user.updateCosmetic();
-        }, 4);
+            Bukkit.getScheduler().runTaskLater(HMCCosmeticsPlugin.getInstance(), () -> {
+                user.spawnBalloon((CosmeticBalloonType) user.getCosmetic(CosmeticSlot.BALLOON));
+                user.updateCosmetic();
+            }, 4);
+        }
     }
 
     @EventHandler