From 2cd00a5b77761da20a00acbeb9c23f3e8c213e7f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Samuel=20=C5=A0tancl?= <samuel.stancl@gmail.com>
Date: Fri, 4 Sep 2020 12:42:08 +0200
Subject: [PATCH] Fix initialize() method

---
 src/Tenancy.php | 23 +++++++++++------------
 1 file changed, 11 insertions(+), 12 deletions(-)

diff --git a/src/Tenancy.php b/src/Tenancy.php
index 0e80221a..8bad1417 100644
--- a/src/Tenancy.php
+++ b/src/Tenancy.php
@@ -31,6 +31,16 @@ class Tenancy
      */
     public function initialize($tenant): void
     {
+        if (! is_object($tenant)) {
+            $tenant = $tenant;
+            $tenantId = $tenant;
+            $tenant = $this->find($tenantId);
+
+            if (! $tenant) {
+                throw new TenantCountNotBeIdentifiedById($tenantId);
+            }
+        }
+
         if ($this->initialized && $this->tenant->getTenantKey() === $tenant->getTenantKey()) {
             return;
         }
@@ -40,18 +50,7 @@ class Tenancy
             $this->end();
         }
 
-        if (is_object($tenant)) {
-            $this->tenant = $tenant;
-        } else {
-            $tenantId = $tenant;
-            $tenant = $this->find($tenantId);
-
-            if (! $tenant) {
-                throw new TenantCountNotBeIdentifiedById($tenantId);
-            }
-
-            $this->tenant = $tenant;
-        }
+        $this->tenant = $tenant;
 
         event(new Events\InitializingTenancy($this));