make code nicer

Change-Id: Ib4d1784f280e761ba22ee723954c35084b041d3e
Signed-off-by: Grarak <williye97@gmail.com>