Added missing return to docs