Trajectory visualization fix