Add doc
Fix broken test