Parametrise this return type too

esoterictemplates · Aug 13, 2024 · e5e542b · e5e542b
1 parent 34d43d4
commit e5e542b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/net/slqmy/template_paper_plugin/custom_entity/AbstractCustomEntity.java b/src/main/java/net/slqmy/template_paper_plugin/custom_entity/AbstractCustomEntity.java
@@ -29,7 +29,7 @@ protected AbstractCustomEntity(TemplatePaperPlugin plugin, CustomEntityManager c
 
   protected abstract E generateCustomEntity(E baseEntity);
 
-  public Entity getCustomEntity(Location spawnLocation) {
+  public E getCustomEntity(Location spawnLocation) {
     E entity = (E) spawnLocation.getWorld().spawnEntity(spawnLocation, entityType);
     entity.getPersistentDataContainer().set(plugin.getCustomEntityIdKey(), PersistentDataType.STRING, entityId.name());
     return generateCustomEntity(entity);