Fix incorrect type assumption