Updated the test to use FlatSpec.