Omega step correction should be column-wise instead of row-wise

[denklewer]

Correct by norm should be performed columnwise for Omega but was removed accidentally.
We should restore it and check