Fix test

Add edited the test for the documentation... forgot to revert my changes.

Signed-off-by: Clement Escoffier <clement.escoffier@gmail.com>