Small fix and check output tests.