--- a/src/internet-node/ipv4-l4-demux.cc	Thu May 10 07:51:59 2007 +0200
+++ b/src/internet-node/ipv4-l4-demux.cc	Thu May 10 07:58:18 2007 +0200
@@ -32,12 +32,10 @@
 
 const Iid Ipv4L4Demux::iid ("Ipv4L4Demux");
 
-Ipv4L4Demux::Ipv4L4Demux (Node *node)
+Ipv4L4Demux::Ipv4L4Demux (Ptr<Node> node)
   : NsUnknown (Ipv4L4Demux::iid),
     m_node (node)
-{
-  m_node->Ref ();
-}
+{}
 
 Ipv4L4Demux::~Ipv4L4Demux()
 {}
@@ -51,11 +49,7 @@
       (*i)->Unref ();
     }
   m_protocols.clear ();
-  if (m_node != 0)
-    {
-      m_node->Unref ();
-      m_node = 0;
-    }
+  m_node = 0;
   NsUnknown::DoDispose ();
 }
 

--- a/src/internet-node/ipv4-l4-demux.h	Thu May 10 07:51:59 2007 +0200
+++ b/src/internet-node/ipv4-l4-demux.h	Thu May 10 07:58:18 2007 +0200
@@ -27,6 +27,7 @@
 
 #include <list>
 #include "ns3/ns-unknown.h"
+#include "ns3/ptr.h"
 
 namespace ns3 {
 
@@ -43,7 +44,7 @@
 public:
   static const Iid iid;
   typedef int Ipv4L4ProtocolTraceType;
-  Ipv4L4Demux (Node *node);
+  Ipv4L4Demux (Ptr<Node> node);
   virtual ~Ipv4L4Demux();
 
   /**
@@ -86,7 +87,7 @@
   virtual void DoDispose (void);
   typedef std::list<Ipv4L4Protocol*> L4List_t;
   L4List_t m_protocols;
-  Node *m_node;
+  Ptr<Node> m_node;
 };
 
 } //namespace ns3