Updates tests to match actual output