Fix minor dictionary key error