Slight clean up of ret code"