Fixed branch length calculation bug

The first omega was used instead of further omegas when calculating
branch lengths.