Fix keys tests