Fix jQuery element wrapping