Changed the way to use a test to make it less verbose.