Small tweak from review.