Updated unit test to match new output.

CI now building fully. Nice.