Documentation figure fix