Fix capacity assignment