diff --git a/day_06/1.rb b/day_06/1.rb
index 62361f9..e347b0e 100755
--- a/day_06/1.rb
+++ b/day_06/1.rb
@@ -40,7 +40,7 @@ class Main
     when :*
       nums.unshift(num1 * num2)
     when :/
-      if nums[1] == 0
+      if num2 == 0
         raise "Division by zero"
       end
       nums.unshift(num1 / num2)
diff --git a/day_06/2.rb b/day_06/2.rb
index ecae48f..4dbf25f 100755
--- a/day_06/2.rb
+++ b/day_06/2.rb
@@ -55,7 +55,7 @@ class Main
     when :*
       nums.unshift(num1 * num2)
     when :/
-      if nums[1] == 0
+      if num2 == 0
         raise "Division by zero"
       end
       nums.unshift(num1 / num2)