1 files changed, 12 insertions, 3 deletions

diff --git a/src/pp.c b/src/pp.c
index 6664879..db37118 100644
--- a/src/pp.c
+++ b/src/pp.c
@@ -75,9 +75,18 @@ void pp_do_com(pp_t* pp) {
 
 void pp_run(pp_t* pp) {
    while (*pp->src) {
-      if (pp->state == PP_STATE_REG) { pp_do_reg(pp); }
-      else if (pp->state == PP_STATE_STR || pp->state == PP_STATE_ESC) { pp_do_str(pp); }
-      else if (pp->state == PP_STATE_COM) { pp_do_com(pp); }
+      switch (pp->state) {
+         case PP_STATE_REG:
+            pp_do_reg(pp);
+            break;
+         case PP_STATE_STR:
+         case PP_STATE_ESC:
+            pp_do_str(pp);
+            break;
+         case PP_STATE_COM:
+            pp_do_com(pp);
+            break;
+      }
 
       pp->src ++;
    }