Fix the key test