Fix copypaste error
Implement SetDesiredCapacity

Fixes TOOLS-206