Fix incorrect key type