From 40b612951c156a5310e90805718f0109e3ad2371 Mon Sep 17 00:00:00 2001
From: brettlangdon <me@brett.is>
Date: Fri, 24 Aug 2018 20:06:23 -0400
Subject: [PATCH] Fix function decorator

---
 importhook/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/importhook/__init__.py b/importhook/__init__.py
index eef8786..95b93a1 100644
--- a/importhook/__init__.py
+++ b/importhook/__init__.py
@@ -68,8 +68,10 @@ def on_import(module_name, func=None):
         importhook.on_import('httplib', on_httplib_import)
     """
     if func is None:
+        @functools.wraps(func)
         def decorator(func):
             registry[module_name] = func
+            return func
         return decorator
     else:
         registry[module_name] = func