Small code style update