small fix code style