High revs before shifting

[Shaneod]

Hi all, I have just had the head gasket done on my a6 c7 and when I started driving it again I noticed that when you floor the accelerator it drops to fourth gear and the revs will go to about 4000rpm before shifting which it never done before, in manual mode it shifts no problem, the battery was disconnected for around a week also, will it just take time for the car to relearn the shift pattern or is this another problem?