As the baby starts to lose balance and moves around all you have to do is hold down spacebar until the baby has gotten its balance back, however you cannot hold down spacebar for too long or the baby will fall over.

Hope this helped, have fun :)