Fix the regex based on review comment.