Fixed bug with return indentation