Fix return command