Slight amend to the regex