Use correct expected call count