Minor change based on PR comments.