Improvement to make removeClass() pass.

adamwathan · May 30, 2015 · 38a7a9c · 38a7a9c
1 parent 77975a9
commit 38a7a9c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/AdamWathan/Form/Elements/Element.php b/src/AdamWathan/Form/Elements/Element.php
@@ -55,7 +55,13 @@ public function removeClass($class)
         if (! isset($this->attributes['class'])) {
             return $this;
         }
+
         $class = trim(str_replace($class, '', $this->attributes['class']));
+        if ($class == '') {
+            $this->removeAttribute('class');
+            return $this;
+        }
+
         $this->setAttribute('class', $class);
         return $this;
     }