Fix peer close test.

Change-Id: Ia1ec88e004683419cf0dd23ef26a49e04e662896