Fix tests to match actual func response structure