Also fix the docblock