2 files changed, 11 insertions, 8 deletions
diff --git a/src/kern/linux/printk.cc b/src/kern/linux/printk.cc
index 7a4c551ce..c2cf60964 100644
--- a/src/kern/linux/printk.cc
+++ b/src/kern/linux/printk.cc
@@ -34,6 +34,7 @@
 #include <algorithm>
 
 #include "base/trace.hh"
+#include "cpu/thread_context.hh"
 #include "kern/linux/printk.hh"
 #include "sim/arguments.hh"
 
diff --git a/src/sim/arguments.hh b/src/sim/arguments.hh
index 58a43852c..165880095 100644
--- a/src/sim/arguments.hh
+++ b/src/sim/arguments.hh
@@ -32,8 +32,8 @@
 #define __SIM_ARGUMENTS_HH__
 
 #include <cassert>
+#include <memory>
 
-#include "base/refcnt.hh"
 #include "base/types.hh"
 #include "mem/fs_translating_port_proxy.hh"
 
@@ -47,7 +47,7 @@ class Arguments
     uint64_t getArg(uint16_t size = (uint16_t)(-1), bool fp = false);
 
   protected:
-    class Data : public RefCounted
+    class Data
     {
       public:
         Data(){}
@@ -60,12 +60,12 @@ class Arguments
         char *alloc(size_t size);
     };
 
-    RefCountingPtr<Data> data;
+    std::shared_ptr<Data> data;
 
   public:
     Arguments(ThreadContext *ctx, int n = 0)
-        : tc(ctx), number(n), data(NULL)
-        { assert(number >= 0); data = new Data;}
+        : tc(ctx), number(n), data(new Data())
+    { assert(number >= 0); }
     Arguments(const Arguments &args)
         : tc(args.tc), number(args.number), data(args.data) {}
     ~Arguments() {}
@@ -73,9 +73,11 @@ class Arguments
     ThreadContext *getThreadContext() const { return tc; }
 
     const Arguments &operator=(const Arguments &args) {
-        tc = args.tc;
-        number = args.number;
-        data = args.data;
+        if (this != &args) {
+            tc = args.tc;
+            number = args.number;
+            data = args.data;
+        }
         return *this;
     }