Minor update to change the key name