Fix peer close test.

Change-Id: I16b0b6f6135a6c1cc8d122308bc0194abe52013f