Update unit test with new regex