Correction to return function.