cherrypick-1.1: storage: Fix tracking of keys written in replica.writeStats