fix doc tests