Fix beam v2 normalization