fix tests for keys