Fixed the unit test docs

Needed this to get the unit tests.