X-Git-Url: http://xvm.mit.edu/gitweb/invirt/third/libt4.git/blobdiff_plain/02967a43024ce81912cd1ec96a800397457f8066..ab9eee5d7f1fbe7a3fe6229d4a78136efb14371b:/rsm_tester.cc

diff --git a/rsm_tester.cc b/rsm_tester.cc
index 5507f80..42aaebe 100644
--- a/rsm_tester.cc
+++ b/rsm_tester.cc
@@ -2,9 +2,9 @@
 // RSM test client
 //
 
-#include "types.h"
-#include "rsm_protocol.h"
-#include "rsmtest_client.h"
+#include "include/types.h"
+#include "include/rsm_protocol.h"
+#include "include/rsmtest_client.h"
 
 int main(int argc, char *argv[]) {
     global = new t4_state('t');
@@ -15,13 +15,11 @@ int main(int argc, char *argv[]) {
 
     rsmtest_client *lc = new rsmtest_client(argv[1]);
     string command(argv[2]);
-    if (command == "partition") {
+    if (command == "partition")
         LOG_NONMEMBER << "net_repair returned " << lc->net_repair(std::stoi(argv[3]));
-    } else if (command == "breakpoint") {
-        int b = std::stoi(argv[3]);
-        LOG_NONMEMBER << "breakpoint " << b << " returned " << lc->breakpoint(b);
-    } else {
+    else if (command == "breakpoint")
+        LOG_NONMEMBER << "breakpoint " << argv[3] << " returned " << lc->breakpoint(std::stoi(argv[3]));
+    else
         LOG_NONMEMBER << "Unknown command " << argv[2];
-    }
     return 0;
 }