Fix straight direction issue