Fix incorrect typing in above commit