fix return type in docs