Fix mistake with dictionary key