Fix docummentation bug