fix keys test to reflect latest change