Fixed compilation error for key_tests