Fixing drifting bug in wheel mode