From 6a2c71bad77ef4b9f60f5ac2921b8a4ff16b4d2d Mon Sep 17 00:00:00 2001
From: Malaber <32635600+Malaber@users.noreply.github.com>
Date: Thu, 2 May 2019 11:12:57 +0200
Subject: [PATCH] Fix stage.team method not returning teams for playoff stages

---
 app/models/stage.rb | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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