Fixing doc for tests
Fixing tests