small fix for beam number handling