Fix missing key test expectations