[minor] Small stying upgrade