Improve document test