a little more modification

Change-Id: I847380712e3375ab1dee3896aabee2ad048232a9