how do I make an item that can open trial vaults? [solved]

This is a bit big but it does work,

BlockPos blockPos = BlockPos.containing(x, y, z);
BlockState blockState = world.getBlockState(blockPos);
if(entity instanceof Player player) {
	if(blockState.getValue(VaultBlock.STATE) == VaultState.ACTIVE) {
		if(world instanceof ServerLevel serverLevel) {
			BlockEntity blockEntity = serverLevel.getBlockEntity(blockPos);
			if(blockEntity instanceof VaultBlockEntity vaultBlockEntity) {
				List<ItemStack> keys = List.of(Items.TRIAL_KEY.getDefaultInstance(), Items.OMINOUS_TRIAL_KEY.getDefaultInstance());
				for(ItemStack itemStack : keys) {
					VaultBlockEntity.Server.tryInsertKey(serverLevel,
							blockPos, blockState,
							vaultBlockEntity.getConfig(),
							vaultBlockEntity.getServerData(),
							vaultBlockEntity.getSharedData(),
							player, itemStack
					);
				}
			}
		}
	}
}