Docblock fix return type