tests: Fix output model oracles (not really)