Fixes return type documentation