Fix return command because of puppet module