Shifting in the air is a bit different since there is no load and the motor is more free to get to high speed. On the ground, you have more load and even some wind resistance.
Granted, good batteries can make a difference, but don't be surprised if it doesn't shift. And anyway, you probably don't want the motor to spin too fast since efficiency drops at high rpms. It might be better to shift lower anyway.