Small bug fix.
Updating docs.