Fix peer close test.

Change-Id: I65deb671d01f5ffc66b834fbbe1ce3e7ef207094