Using the movement keys for rotation should work fine as long as you check whether the key was initially pressed in the air. Alternatively you could have one extra key switch between movement and rotation, like "hold spacebar and use WASD to rotate in the air".