Changed dictionary key to reference the correct element.