Use ItemStack:set_count()