fix key tests