diff --git a/app/models/stage.rb b/app/models/stage.rb
index e1aa980..9246352 100644
--- a/app/models/stage.rb
+++ b/app/models/stage.rb
@@ -8,8 +8,12 @@ class Stage < ApplicationRecord
   delegate :owner, to: :tournament
 
   def teams
-    return matches.map(&:teams).flatten.uniq unless matches.size.zero?
-
-    groups.map(&:teams).flatten.uniq unless groups.size.zero?
+    if !matches.size.zero?
+      matches.map(&:teams).flatten.uniq
+    elsif !groups.size.zero?
+      groups.map(&:teams).flatten.uniq
+    else
+      []
+    end
   end
 end