Fixed wrapAll function