function made more elegant

2 files changed, 5 insertions, 8 deletions

diff --git a/exercises/048_methods2.zig b/exercises/048_methods2.zig
index 310867a..3291965 100644
--- a/exercises/048_methods2.zig
+++ b/exercises/048_methods2.zig
@@ -52,12 +52,9 @@ fn visitElephants(first_elephant: *Elephant) void {
         e.print();
         e.visit();
 
-        // This gets the next elephant or stops.
-        if (e.hasTail()) {
-            e = e.???; // Which method do we want here?
-        } else {
-            break;
-        }
+        // This gets the next elephant or stops:
+        // which method do we want here?
+        e = if (e.hasTail()) e.??? else break;
     }
 }
 
diff --git a/patches/patches/048_methods2.patch b/patches/patches/048_methods2.patch
index 781a99e..cd1b5d0 100644
--- a/patches/patches/048_methods2.patch
+++ b/patches/patches/048_methods2.patch
@@ -1,4 +1,4 @@
 57c57
-<             e = e.???; // Which method do we want here?
+<         e = if (e.hasTail()) e.??? else break;
 ---
->             e = e.getTail(); // Which method do we want here?
+>         e = if (e.hasTail()) e.getTail() else break;