Fix return line