Minor tweaks to the Key Discovery API tests.