correct code style

correct code style