Use correct dictionary key