From ec08ddc59e03baf3ab2d868a5f986d679fe3ef6f Mon Sep 17 00:00:00 2001
From: Tim Neutkens <timneutkens@me.com>
Date: Tue, 24 Oct 2017 10:37:28 +0200
Subject: [PATCH] Add website code to body class

---
 Plugin/StoreCodeBodyClassPlugin.php | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/Plugin/StoreCodeBodyClassPlugin.php b/Plugin/StoreCodeBodyClassPlugin.php
index 47fc05e..dbbe87d 100644
--- a/Plugin/StoreCodeBodyClassPlugin.php
+++ b/Plugin/StoreCodeBodyClassPlugin.php
@@ -27,7 +27,10 @@ public function __construct(
     }
 
     public function execute(Observer $observer){
-        $storeCode = $this->storeManager->getStore()->getCode();
+        $store = $this->storeManager->getStore();
+        $storeCode = $store->getCode();
+        $websiteCode = $store->getWebsite()->getCode();
         $this->config->addBodyClass($storeCode);
+        $this->config->addBodyClass($websiteCode);
     }
 }
\ No newline at end of file