Fixes stage alignment