fix doc and tests