Fixes doc block return