Added missing key_name parameter

I forget to add key_name parameter in two function, kindly add these fix