diff --git a/src/DI/ProxyExtension.php b/src/DI/ProxyExtension.php
index 2d2e130..d2450b4 100644
--- a/src/DI/ProxyExtension.php
+++ b/src/DI/ProxyExtension.php
@@ -103,7 +103,7 @@ public function beforeCompile()
 				$def->setTags(array_diff_key($def->getTags(), [self::TAG_LAZY => null]));
 				continue;
 			}
-			$def->addTag(self::TAG_LAZY, $def->getImplement() ?: $def->getClass());
+			$def->addTag(self::TAG_LAZY, $def->getType());
 		}
 	}