Fix return indentation