fix doc + add test