Fix incorrect assertion