Update docs
Small bug fix