Fix incorrect type