How to make the player pick up the block they break

this works but it might be a bit jank

!((Player)entity).hasCorrectToolForDrops(blockstate, (ServerLevel)world, BlockPos.containing(x, y, z))

		
		if (entity instanceof Player _player) {
			List<ItemStack> items = Block.getDrops(blockstate, (ServerLevel)world, BlockPos.containing(x, y, z), world
					.getBlockEntity(BlockPos.containing(x, y, z)), _player, _player.getMainHandItem());

			for(ItemStack item: items) {
				_player.addItem(item);
			}
		}