diff --git a/css/workflows-field-diagram.css b/css/workflows-field-diagram.css
index f87ab31..24984f7 100644
--- a/css/workflows-field-diagram.css
+++ b/css/workflows-field-diagram.css
@@ -1,5 +1,5 @@
 .mermaid g.node.active-state > rect {
-  fill: #D5E8D4 !important;
-  stroke: #82B366 !important;
+  fill: #d5e8d4 !important;
+  stroke: #82b366 !important;
   stroke-width: 3px !important;
 }
diff --git a/src/Plugin/Field/FieldFormatter/WorkflowsFieldDiagramFormatter.php b/src/Plugin/Field/FieldFormatter/WorkflowsFieldDiagramFormatter.php
index fb7b14a..252fa62 100644
--- a/src/Plugin/Field/FieldFormatter/WorkflowsFieldDiagramFormatter.php
+++ b/src/Plugin/Field/FieldFormatter/WorkflowsFieldDiagramFormatter.php
@@ -25,8 +25,8 @@ class WorkflowsFieldDiagramFormatter extends FormatterBase {
    */
   public static function defaultSettings() {
     return [
-        'flowchart-direction' => 'TD',
-      ] + parent::defaultSettings();
+      'flowchart-direction' => 'TD',
+    ] + parent::defaultSettings();
   }
 
   /**
@@ -66,18 +66,17 @@ class WorkflowsFieldDiagramFormatter extends FormatterBase {
   public function viewElements(FieldItemListInterface $items, $langcode) {
     $element = [];
 
-    // load workflow
+    // Load workflow.
     $workflow_id = $this->fieldDefinition->getSetting('workflow');
-    $workflow = Workflow::load($workflow_id);
-    $workflowType = $workflow->getTypePlugin();
-    $states = $workflowType->getStates();
+    $workflow = $this->($workflow_id)->getTypePlugin()->getStates();
+  
 
     foreach ($items as $delta => $item) {
 
-      // mark active state
+      // Mark active state.
       $active_state_id = $item->value;
 
-      // pass classes to states
+      // Pass classes to states.
       $classes = [
         'states' => [
           $active_state_id => ['active-state'],
@@ -85,7 +84,7 @@ class WorkflowsFieldDiagramFormatter extends FormatterBase {
         'transitions' => [],
       ];
 
-      // disable states or transitions,
+      // Disable states or transitions,.
       $disabled = [
         'states' => [],
         'transitions' => [],