Make doc test work