Fix return lines