Fixed bug in capacity choice menu