diff --git a/diagrams/__init__.py b/diagrams/__init__.py
index 74237767..af40a6c0 100644
--- a/diagrams/__init__.py
+++ b/diagrams/__init__.py
@@ -37,6 +37,10 @@ def getcluster():
 def setcluster(cluster):
     __cluster.set(cluster)
 
+def new_init(cls, init):
+    def reset_init(*args, **kwargs):
+        cls.__init__ = init
+    return reset_init
 def new_init(cls, init):
     def reset_init(*args, **kwargs):
         cls.__init__ = init