Fix keys tests/add key button