Fix minor code style