Fixed breaking keys tests