Fixes stage alignment
Merge pull request #892 from yurydelendik/stage-align-fix

Fixes stage alignment