Basic unit test

2 files changed, 2 insertions, 2 deletions

diff --git a/lib/wmi_lite/wmi.rb b/lib/wmi_lite/wmi.rb
index 4a30573..57c1ab2 100644
--- a/lib/wmi_lite/wmi.rb
+++ b/lib/wmi_lite/wmi.rb
@@ -62,7 +62,7 @@ module WmiLite
     end
 
     def wmi_result_to_snapshot(wmi_object)
-      snapshot = Instance.wmi_ole_object_to_hash(wmi_object)
+      snapshot = Instance.new(wmi_object)
     end
   end
 end
diff --git a/lib/wmi_lite/wmi/wmi_instance.rb b/lib/wmi_lite/wmi/wmi_instance.rb
index d0b0474..88228c7 100644
--- a/lib/wmi_lite/wmi/wmi_instance.rb
+++ b/lib/wmi_lite/wmi/wmi_instance.rb
@@ -38,7 +38,7 @@ module WmiLite
           property_map[property.name.downcase] = wmi_object.invoke(property.name)
         end
 
-        property_map[:wmi_object] = wmi_object
+        @wmi_ole_object = wmi_object
 
         property_map.freeze
       end