chore(test): Add Key handling