Fix documentation tests