net/mlx5_core: Fix a bug in alloc_token