diff --git a/src/collector/InheritanceResolver.php b/src/collector/InheritanceResolver.php
index 814774fc..456489d0 100644
--- a/src/collector/InheritanceResolver.php
+++ b/src/collector/InheritanceResolver.php
@@ -162,7 +162,7 @@ private function processExtends(AbstractUnitObject $unit, AbstractUnitObject $ex
                         $extendedUnit = $this->getUnitByName($name);
                         $this->processExtends($unit, $extendedUnit, $extendedUnit);
                     } catch (ProjectException $e) {
-                        $this->addUnresolved($unit, $extends->getExtends());
+                        $this->addUnresolved($unit, $name);
                     }
                 }
             }
@@ -227,7 +227,7 @@ private function processTraitUse(AbstractUnitObject $unit, TraitUseObject $use,
                         $extendedUnit = $this->getUnitByName($name);
                         $this->processExtends($unit, $extendedUnit, $extendedUnit);
                     } catch (ProjectException $e) {
-                        $this->addUnresolved($unit, $trait->getExtends());
+                        $this->addUnresolved($unit, $name);
                     }
                 }
             }