Add wraps call around class decorator in inject

1 files changed, 2 insertions, 2 deletions

diff --git a/di/core.py b/di/core.py
index db3202e..f41d601 100644
--- a/di/core.py
+++ b/di/core.py
@@ -133,8 +133,8 @@ def inject(*args, **kwargs):
             return obj
         
         if not isinstance(obj, DiRegistry):
-            obj = DiRegistry(obj)
-
+            obj = wraps(obj)(DiRegistry(obj))
+            
         obj.register(*args, **kwargs)
         return obj