Fix doc tests