Fix incorrect dict setting

Thanks again @itchyny!