1 files changed, 4 insertions, 4 deletions
diff --git a/src/main/java/net/girlonthe/chattweaks/mixin/ChatComponentMixin.java b/src/main/java/net/girlonthe/chattweaks/mixin/ChatComponentMixin.java
index bdbc428..a228b10 100644
--- a/src/main/java/net/girlonthe/chattweaks/mixin/ChatComponentMixin.java
+++ b/src/main/java/net/girlonthe/chattweaks/mixin/ChatComponentMixin.java
@@ -2,7 +2,7 @@ package net.girlonthe.chattweaks.mixin;
 
 import net.minecraft.ChatFormatting;
 import net.minecraft.client.gui.components.ChatComponent;
-import net.minecraft.client.GuiMessage;
+import net.minecraft.client.multiplayer.chat.GuiMessage;
 import net.minecraft.network.chat.Component;
 import net.minecraft.network.chat.Style;
 import net.minecraft.network.chat.TextColor;
@@ -16,8 +16,8 @@ import java.time.format.DateTimeFormatter;
 @Mixin(ChatComponent.class)
 public abstract class ChatComponentMixin {
     @ModifyArg(
-        method = "addMessage(Lnet/minecraft/network/chat/Component;Lnet/minecraft/network/chat/MessageSignature;Lnet/minecraft/client/GuiMessageTag;)V",
-        at = @At(value = "INVOKE", target = "Lnet/minecraft/client/gui/components/ChatComponent;addMessageToDisplayQueue(Lnet/minecraft/client/GuiMessage;)V")
+        method = "addMessage",
+        at = @At(value = "INVOKE", target = "Lnet/minecraft/client/gui/components/ChatComponent;addMessageToDisplayQueue(Lnet/minecraft/client/multiplayer/chat/GuiMessage;)V")
     )
     private GuiMessage rosasChatTweaks$prependTimestamp(GuiMessage original) {
         Component newComponent = Component.empty()
@@ -26,6 +26,6 @@ public abstract class ChatComponentMixin {
                     .setStyle(Style.EMPTY.withColor(ChatFormatting.GRAY))
             )
             .append(original.content());
-        return new GuiMessage(original.addedTime(), newComponent, original.signature(), original.tag());
+        return new GuiMessage(original.addedTime(), newComponent, original.signature(), original.source(), original.tag());
     }
 }