Fixing up the use_key function.