Updated test for output testing.