fix: use WrapOperation for MusicManagerMixin

Author: bconlon1

common/src/main/java/com/aetherteam/cumulus/mixin/mixins/client/MusicManagerMixin.java

@@ -1,23 +1,24 @@
 package com.aetherteam.cumulus.mixin.mixins.client;
 
 import com.aetherteam.cumulus.client.CumulusClient;
+import com.llamalad7.mixinextras.injector.wrapoperation.Operation;
+import com.llamalad7.mixinextras.injector.wrapoperation.WrapOperation;
 import net.minecraft.client.Minecraft;
 import net.minecraft.client.sounds.MusicManager;
 import net.minecraft.sounds.Music;
 import net.minecraft.sounds.Musics;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.injection.At;
-import org.spongepowered.asm.mixin.injection.ModifyVariable;
 
 @Mixin(MusicManager.class)
 public class MusicManagerMixin {
     /**
      * This mixin modifies the return of {@link Minecraft#getSituationalMusic()} as it is given to the music variable in {@link MusicManager}.
-     * @param music The original {@link Music} variable.
      * @return The modified {@link Music} variable.
      */
-    @ModifyVariable(method = "tick()V", at = @At(value = "STORE"))
-    public Music injected(Music music) {
+    @WrapOperation(method = "tick()V", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/Minecraft;getSituationalMusic()Lnet/minecraft/sounds/Music;"))
+    public Music injected(Minecraft instance, Operation<Music> original) {
+        Music music = original.call(instance);
         if (music == Musics.MENU && CumulusClient.MENU_HELPER.getActiveMusic() != null) {
             return CumulusClient.MENU_HELPER.getActiveMusic();
         }