Fix documentation and broken test