diff --git a/src/main/java/tschipp/carryon/common/capabilities/PositionProvider.java b/src/main/java/tschipp/carryon/common/capabilities/PositionProvider.java index cbd7baa..b4ada10 100644 --- a/src/main/java/tschipp/carryon/common/capabilities/PositionProvider.java +++ b/src/main/java/tschipp/carryon/common/capabilities/PositionProvider.java @@ -7,18 +7,23 @@ import net.minecraftforge.common.capabilities.CapabilityInject; import net.minecraftforge.common.capabilities.ICapabilitySerializable; import net.minecraftforge.common.util.LazyOptional; -public class PositionProvider implements ICapabilitySerializable { +public class PositionProvider implements ICapabilitySerializable +{ @CapabilityInject(IPosition.class) public static final Capability POSITION_CAPABILITY = null; - + private IPosition instance = POSITION_CAPABILITY.getDefaultInstance(); @SuppressWarnings("unchecked") @Override public LazyOptional getCapability(Capability cap, Direction side) { - return (LazyOptional) LazyOptional.of(() -> {return new TEPosition();}); + if (cap == POSITION_CAPABILITY) + return (LazyOptional) LazyOptional.of(() -> { + return new TEPosition(); + }); + return null; } @Override @@ -30,7 +35,7 @@ public class PositionProvider implements ICapabilitySerializable { @Override public void deserializeNBT(CompoundNBT nbt) { - POSITION_CAPABILITY.getStorage().readNBT(POSITION_CAPABILITY, instance, null, nbt); + POSITION_CAPABILITY.getStorage().readNBT(POSITION_CAPABILITY, instance, null, nbt); } - + }