Make test more awesome with regex