Fixing test with failing output builder