--- a/samples/main-simple.cc	Wed May 02 23:45:26 2007 +0200
+++ b/samples/main-simple.cc	Thu May 03 00:12:39 2007 +0200
@@ -17,6 +17,10 @@
     {
       Simulator::Schedule (Seconds (0.5), &GenerateTraffic, socket, size - 50);
     }
+  else
+    {
+      socket->Close ();
+    }
 }
 
 static void
@@ -35,12 +39,17 @@
 {
   InternetNode *a = new InternetNode ();
 
-  Socket *sink = a->GetUdp ()->CreateSocket ();
+  Udp *udp;
+  udp = a->GetUdp ();
+
+  Socket *sink = udp->CreateSocket ();
   sink->Bind (80);
 
-  Socket *source = a->GetUdp ()->CreateSocket ();
+  Socket *source = udp->CreateSocket ();
   source->Connect (Ipv4Address::GetLoopback (), 80);
 
+  udp->Unref ();
+
   GenerateTraffic (source, 500);
   PrintTraffic (sink);
 
@@ -49,10 +58,9 @@
 
   Simulator::Destroy ();
 
-
-  delete a;
-  delete source;
-  delete sink;
+  sink->Unref ();
+  source->Unref ();
+  a->Unref ();
 
   return 0;
 }