Add wave direction correction