diff --git a/setup/__init__.py b/setup/__init__.py
new file mode 100644
index 0000000..225fa23
--- /dev/null
+++ b/setup/__init__.py
@@ -0,0 +1,9 @@
+import sys
+
+if sys.version_info >= ( 3, 0 ):
+    from pykd.pykd import *
+else:
+    from pykd import *
+
+__version__ = pykd.__version__
+__file__ = pykd.__file__
diff --git a/setup/setup.py b/setup/setup.py
index ffabeca..07cddc8 100644
--- a/setup/setup.py
+++ b/setup/setup.py
@@ -27,10 +27,7 @@ def  makeWheel(args):
         shutil.rmtree(package_dir)
     os.mkdir(package_dir)
 
-    with open(os.path.join(package_dir, '__init__.py'),'w') as f:
-        f.write("from pykd import *\n")
-        f.write("__version__ = pykd.__version__\n")
-        f.write("__file__ = pykd.__file__\n")
+    shutil.copy("__init__.py", package_dir)
 
     bin_dir = os.path.join( os.path.curdir, '..', 'bin')
     pykd_dir =  os.path.join( os.path.curdir, '..', 'out')