One bug fix

Small bug fix and documentation fix