Control of the pump unit