fix test broke by total calculation