diff --git a/lib/sparkr/sparkline.rb b/lib/sparkr/sparkline.rb
index 2fff792..55d9e71 100644
--- a/lib/sparkr/sparkline.rb
+++ b/lib/sparkr/sparkline.rb
@@ -46,16 +46,10 @@ def initialize(_numbers)
     #   # => "▁█" (colored, which you can't see)
     #
     # @return [Sparkline] itself
-    def format(&block)
-      new_ticks = []
-      @ticks.each_with_index do |tick, index|
-        if block.arity == 2
-          new_ticks << yield(tick, @original_numbers[index])
-        elsif block.arity == 3
-          new_ticks << yield(tick, @original_numbers[index], index)
-        end
+    def format
+      @ticks = @ticks.map.with_index do |tick, index|
+        yield tick, @original_numbers[index], index
       end
-      @ticks = new_ticks
       self
     end