Some review comment fixes

Change-Id: Ib9f2048617379eb17e207a16176d0e77b0d37d6e