Clean up execution key function
Use correct input name