Fix tests
Remove docs for test contexts (will be deprecated soon)