public LivingPet getPet(Entity pet) {
for (LivingPet pi : pets) {
if (pi.getEntityPet().equals(pet) || pi.getMount().getEntityPet().equals(pet)) {
return pi;
}
if (pi.getCraftPet().equals(pet) || pi.getMount().getCraftPet().equals(pet)) {
return pi;
}
}
return null;
}