Change the set key function.