Fix incorrect code