diff --git a/src/main/java/baritone/bot/IBaritoneUser.java b/src/main/java/baritone/bot/IBaritoneUser.java
index 1ef6cf9b9..50ab8783d 100644
--- a/src/main/java/baritone/bot/IBaritoneUser.java
+++ b/src/main/java/baritone/bot/IBaritoneUser.java
@@ -17,8 +17,8 @@
 
 package baritone.bot;
 
-import baritone.bot.net.BotNetHandlerPlayClient;
 import net.minecraft.client.entity.EntityOtherPlayerMP;
+import net.minecraft.network.NetworkManager;
 import net.minecraft.world.World;
 
 import javax.annotation.Nullable;
@@ -30,9 +30,9 @@ import javax.annotation.Nullable;
 public interface IBaritoneUser {
 
     /**
-     * @return The current connection handler.
+     * @return The network manager that is responsible for the current connection.
      */
-    @Nullable BotNetHandlerPlayClient getConnection();
+    @Nullable NetworkManager getNetworkManager();
 
     /**
      * @return The locally managed entity for this bot.