diff --git a/2016/ruby/day_13.rb b/2016/ruby/day_13.rb
index d0d7bee..c41bd27 100644
--- a/2016/ruby/day_13.rb
+++ b/2016/ruby/day_13.rb
@@ -53,7 +53,7 @@ end
 
 if __FILE__ == $0
   maze = Maze.new(1358)
-  p maze.solve([1,1]).walk.find {|pos,_| pos == [31,39] }
+  p maze.solve([1,1]).walk.take_while {|pos,count| count < 51 }.map(&:first).uniq.size
 end
 
 require 'minitest'