Fix peer close test.

Change-Id: Ie0d4d95fb3d21801ef59844402317f35e88ccbeb