[misc] Minor code improvement