Fix dictionary key error