1 files changed, 8 insertions, 7 deletions
diff --git a/src/mem/ruby/slicc_interface/NetworkMessage.hh b/src/mem/ruby/slicc_interface/NetworkMessage.hh
index 41365ff9e..c92de0937 100644
--- a/src/mem/ruby/slicc_interface/NetworkMessage.hh
+++ b/src/mem/ruby/slicc_interface/NetworkMessage.hh
@@ -31,8 +31,7 @@
 
 #include <iostream>
 
-#include "mem/gems_common/RefCnt.hh"
-#include "mem/gems_common/RefCountable.hh"
+#include "base/refcnt.hh"
 #include "mem/protocol/MessageSizeType.hh"
 #include "mem/ruby/common/Global.hh"
 #include "mem/ruby/common/NetDest.hh"
@@ -41,16 +40,18 @@
 class Address;
 
 class NetworkMessage;
-typedef RefCnt<NetworkMessage> NetMsgPtr;
+typedef RefCountingPtr<NetworkMessage> NetMsgPtr;
 
 class NetworkMessage : public Message
 {
   public:
     NetworkMessage()
-        : Message()
-    {
-        m_internal_dest_valid = false;
-    }
+        : m_internal_dest_valid(false)
+    { }
+
+    NetworkMessage(const NetworkMessage &other)
+        : Message(other), m_internal_dest_valid(other.m_internal_dest_valid)
+    { }
 
     virtual ~NetworkMessage() { }