Slight change to the regex