Помогите с модом forge 1.16.5

[LeCrovVv]

Выдает ошибку Cannot resolve method 'getEntity' in 'Object', строка 19
Версия java: 16

Спойлер

 

package org.weeeirdoo.bmanager;

import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.context.CommandContext;
import net.minecraft.command.CommandSource;
import net.minecraft.entity.Entity;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.util.text.StringTextComponent;

import static com.mojang.brigadier.builder.LiteralArgumentBuilder.literal;

public class BManager {
    public static void register(CommandDispatcher<CommandSource> dispatcher) {
        dispatcher.register(
                literal("whisper")
                        .then(
                                literal("baritone")
                                        .executes(context -> {
                                            Entity entity = context.getSource().getEntity();
                                            if (entity instanceof PlayerEntity) {
                                                PlayerEntity player = (PlayerEntity) entity;
                                                String argument = context.getArgument("argument", String.class);
                                                sendMessageToBaritone(player, "baritone " + argument);
                                            }
                                            return 1;
                                        })
                        )
        );
    }

    private static void sendMessageToBaritone(PlayerEntity player, String message) {
        player.sendMessage(new StringTextComponent("#" + message), player.getUUID());
    }
}