Fix return function.