diff --git a/src/main/java/de/florianmichael/viaforge/mixin/impl/MixinServerData.java b/src/main/java/de/florianmichael/viaforge/mixin/impl/MixinServerData.java
index aea8940..8344531 100644
--- a/src/main/java/de/florianmichael/viaforge/mixin/impl/MixinServerData.java
+++ b/src/main/java/de/florianmichael/viaforge/mixin/impl/MixinServerData.java
@@ -34,7 +34,7 @@ import org.spongepowered.asm.mixin.injection.callback.LocalCapture;
 public class MixinServerData implements ExtendedServerData {
 
     @Unique
-    private VersionEnum viaForge$version;
+    private ProtocolVersion viaForge$version;
 
     @Inject(method = "getNBTCompound", at = @At(value = "INVOKE", target = "Lnet/minecraft/nbt/NBTTagCompound;setString(Ljava/lang/String;Ljava/lang/String;)V", ordinal = 0), locals = LocalCapture.CAPTURE_FAILHARD)
     public void saveVersion(CallbackInfoReturnable<NBTTagCompound> cir, NBTTagCompound nbttagcompound) {
@@ -64,12 +64,12 @@ public class MixinServerData implements ExtendedServerData {
     }
 
     @Override
-    public VersionEnum viaForge$getVersion() {
+    public ProtocolVersion viaForge$getVersion() {
         return viaForge$version;
     }
 
     @Override
-    public void viaForge$setVersion(VersionEnum version) {
+    public void viaForge$setVersion(ProtocolVersion version) {
         viaForge$version = version;
     }
 
diff --git a/src/main/java/de/florianmichael/viaforge/mixin/impl/fixes/MixinEntityPlayerSP.java b/src/main/java/de/florianmichael/viaforge/mixin/impl/fixes/MixinEntityPlayerSP.java
index a76082d..a1ff0cf 100644
--- a/src/main/java/de/florianmichael/viaforge/mixin/impl/fixes/MixinEntityPlayerSP.java
+++ b/src/main/java/de/florianmichael/viaforge/mixin/impl/fixes/MixinEntityPlayerSP.java
@@ -26,7 +26,6 @@ import net.minecraft.client.entity.EntityPlayerSP;
 import net.minecraft.client.network.NetHandlerPlayClient;
 import net.minecraft.network.Packet;
 import net.minecraft.world.World;
-import net.raphimc.vialoader.util.VersionEnum;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.Shadow;
 import org.spongepowered.asm.mixin.Unique;