Fix bug in IAST wrapping