Fixing returns that should be return in documentation