Fix tests
Update doc