diff --git a/app/models/stage.rb b/app/models/stage.rb
index 48ad20d..0beb43b 100644
--- a/app/models/stage.rb
+++ b/app/models/stage.rb
@@ -6,8 +6,12 @@ class Stage < ApplicationRecord
   has_many :groups, dependent: :destroy
 
   def teams
-    return matches.map(&:teams).flatten.uniq unless matches.size.zero?
+    if groups.size.zero?
+      return matches.map(&:teams).flatten.uniq
+    elsif matches.size.zero?
+      return groups.map(&:teams).flatten.uniq
+    end
 
-    groups.map(&:teams).flatten.uniq unless groups.size.zero?
+    []
   end
 end