diff --git a/src/main/java/org/esoteric_organisation/tss_lobby_plugin/cosmetic/hat/HatType.java b/src/main/java/org/esoteric_organisation/tss_lobby_plugin/cosmetic/hat/HatType.java
index 81de8e3..8dc6ad2 100644
--- a/src/main/java/org/esoteric_organisation/tss_lobby_plugin/cosmetic/hat/HatType.java
+++ b/src/main/java/org/esoteric_organisation/tss_lobby_plugin/cosmetic/hat/HatType.java
@@ -7,6 +7,7 @@
 import org.bukkit.Material;
 import org.bukkit.entity.Player;
 import org.bukkit.inventory.ItemStack;
+import org.jetbrains.annotations.NotNull;
 
 public enum HatType implements Cosmetic {
   TOP_HAT(new TranslatableItemStack(Material.PLAYER_HEAD, Message.TOP_HAT_DISPLAY_NAME, Message.TOP_HAT_DESCRIPTION)),
@@ -20,7 +21,7 @@ public enum HatType implements Cosmetic {
   }
 
   @Override
-  public ItemStack getDisplayItem(Player player, TSSLobbyPlugin plugin) {
+  public ItemStack getDisplayItem(Player player, @NotNull TSSLobbyPlugin plugin) {
 	return item.asBukkitItemStack(player, plugin.getCore());
   }
 
diff --git a/src/main/java/org/esoteric_organisation/tss_lobby_plugin/cosmetic/particle_cosmetic/trail/TrailType.java b/src/main/java/org/esoteric_organisation/tss_lobby_plugin/cosmetic/particle_cosmetic/trail/TrailType.java
index 37c6ab8..9d72291 100644
--- a/src/main/java/org/esoteric_organisation/tss_lobby_plugin/cosmetic/particle_cosmetic/trail/TrailType.java
+++ b/src/main/java/org/esoteric_organisation/tss_lobby_plugin/cosmetic/particle_cosmetic/trail/TrailType.java
@@ -8,6 +8,7 @@
 import org.bukkit.Particle;
 import org.bukkit.entity.Player;
 import org.bukkit.inventory.ItemStack;
+import org.jetbrains.annotations.NotNull;
 
 public enum TrailType implements ParticleCosmetic {
   HEART(
@@ -60,7 +61,7 @@ public enum TrailType implements ParticleCosmetic {
   }
 
   @Override
-  public ItemStack getDisplayItem(Player player, TSSLobbyPlugin plugin) {
+  public ItemStack getDisplayItem(Player player, @NotNull TSSLobbyPlugin plugin) {
 	return displayItem.asBukkitItemStack(player, plugin.getCore());
   }