It wasn't really needed here as there are no calls on `wrap()`. I guess for consistency, we should add it here. Can you do that? Thanks