DJI F450 wobble problem

I had a wobble problem right before lift off, once in the air it hovers very nicely. After doing some tests I concluded the wobble is from the landing legs combined with being top heavy with a 5200mah battery.