Add safecheck for getFirstElementChild and replace another instance of firstChild

2 files changed, 8 insertions, 6 deletions

diff --git a/src/Nodes/NodeTrait.php b/src/Nodes/NodeTrait.php
index ccc82ab..9ef1fa2 100644
--- a/src/Nodes/NodeTrait.php
+++ b/src/Nodes/NodeTrait.php
@@ -181,11 +181,11 @@ trait NodeTrait
     /**
      * Override for native hasAttribute.
      *
-     * @see getAttribute
-     *
      * @param $attributeName
      *
      * @return bool
+     * @see getAttribute
+     *
      */
     public function hasAttribute($attributeName)
     {
@@ -566,9 +566,11 @@ trait NodeTrait
      */
     public function getFirstElementChild()
     {
-        foreach ($this->childNodes as $node) {
-            if ($node instanceof \DOMElement) {
-                return $node;
+        if ($this->childNodes instanceof \Traversable) {
+            foreach ($this->childNodes as $node) {
+                if ($node instanceof \DOMElement) {
+                    return $node;
+                }
             }
         }
 
diff --git a/src/Readability.php b/src/Readability.php
index 1c24633..c8a321b 100644
--- a/src/Readability.php
+++ b/src/Readability.php
@@ -1610,7 +1610,7 @@ class Readability
             $node->removeAttribute('class');
         }
 
-        for ($node = $node->firstChild; $node !== null; $node = $node->nextSibling) {
+        for ($node = $node->getFirstElementChild(); $node !== null; $node = $node->nextSibling) {
             $this->_cleanClasses($node);
         }
     }