Correct a unit test and location of 2 doc strings