Fix total capacity issue