diff --git a/swig/pyniflib.i b/swig/pyniflib.i
index c997f3b8f058b31748edb4dccdddbee5190f2b63..5b7b8f369a28e39835bfe46f260c8508d127f005 100644
--- a/swig/pyniflib.i
+++ b/swig/pyniflib.i
@@ -610,6 +610,7 @@ POSSIBILITY OF SUCH DAMAGE. */
 %include "python.h"
 %{
 #include "python.h"
+#include "../include/gen/obj_defines.h"
 %}
 
 %template(vector_byte) std::vector<Niflib::byte>;
@@ -721,6 +722,7 @@ POSSIBILITY OF SUCH DAMAGE. */
 %template(NiBSplineInterpolatorRef) Niflib::Ref<Niflib::NiBSplineInterpolator>;
 %template(DynamicCastToNiBSplineInterpolator) Niflib::DynamicCast<Niflib::NiBSplineInterpolator>;
 %template(StaticCastToNiBSplineInterpolator) Niflib::StaticCast<Niflib::NiBSplineInterpolator>;
+*/
 %include "../include/obj/NiObjectNET.h"
 %template(NiObjectNETRef) Niflib::Ref<Niflib::NiObjectNET>;
 %template(DynamicCastToNiObjectNET) Niflib::DynamicCast<Niflib::NiObjectNET>;
@@ -729,6 +731,7 @@ POSSIBILITY OF SUCH DAMAGE. */
 %template(NiAVObjectRef) Niflib::Ref<Niflib::NiAVObject>;
 %template(DynamicCastToNiAVObject) Niflib::DynamicCast<Niflib::NiAVObject>;
 %template(StaticCastToNiAVObject) Niflib::StaticCast<Niflib::NiAVObject>;
+/*
 %include "../include/obj/NiDynamicEffect.h"
 %template(NiDynamicEffectRef) Niflib::Ref<Niflib::NiDynamicEffect>;
 %template(DynamicCastToNiDynamicEffect) Niflib::DynamicCast<Niflib::NiDynamicEffect>;
diff --git a/swig/pyniflib_test.py b/swig/pyniflib_test.py
index 060398d45413661e0068d9d9ed45d9c40923187a..b7fa938f51e418d01c5d821f32a82cdf563e5734 100644
--- a/swig/pyniflib_test.py
+++ b/swig/pyniflib_test.py
@@ -2,17 +2,13 @@
 
 from niflib import *
 
-NifInfo()
-
-x = Matrix44(1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16)
-print x
-
 root = CreateNiNode()
-print root
 
+x = Matrix44(1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16)
 root.SetLocalTransform(x)
-print root
 
-WriteNifTree( "test.nif", root, NifInfo() )
+print root
 
+#nifinfo = NifInfo()
+#WriteNifTree( "test.nif", root.Ptr(), nifinfo )
 
diff --git a/swig/python.h b/swig/python.h
index a8b184b7f6659a408a2d9a9b56e69c27cfd01011..39791e9d5678f599f14f77521f5d59f3634f7e36 100644
--- a/swig/python.h
+++ b/swig/python.h
@@ -1,2 +1,4 @@
+using namespace Niflib;
+
 #include "../include/obj/NiNode.h"
-NiNodeRef CreateNiNode() { return Ref<NiNode>(new NiNode); }
+NiNodeRef CreateNiNode() { return NiNodeRef(new NiNode); }