4 files changed, 7 insertions, 7 deletions
diff --git a/src/python/m5/SimObject.py b/src/python/m5/SimObject.py
index 240b449df..4a509e57d 100644
--- a/src/python/m5/SimObject.py
+++ b/src/python/m5/SimObject.py
@@ -920,7 +920,7 @@ class SimObject(object):
     def export_methods(cls, code):
         code('''
     void init();
-    void loadState(Checkpoint *cp);
+    void loadState(CheckpointIn &cp);
     void initState();
     void regStats();
     void resetStats();
diff --git a/src/python/swig/core.i b/src/python/swig/core.i
index ed520e17a..862c0e37e 100644
--- a/src/python/swig/core.i
+++ b/src/python/swig/core.i
@@ -86,8 +86,8 @@ Tick curTick();
 class Checkpoint;
 
 void serializeAll(const std::string &cpt_dir);
-Checkpoint *getCheckpoint(const std::string &cpt_dir);
-void unserializeGlobals(Checkpoint *cp);
+CheckpointIn *getCheckpoint(const std::string &cpt_dir);
+void unserializeGlobals(CheckpointIn &cp);
 
 bool want_warn, warn_verbose;
 bool want_info, info_verbose;
diff --git a/src/python/swig/pyobject.cc b/src/python/swig/pyobject.cc
index 08b9ee69f..f90e456e2 100644
--- a/src/python/swig/pyobject.cc
+++ b/src/python/swig/pyobject.cc
@@ -192,8 +192,8 @@ PythonSimObjectResolver::resolveSimObject(const string &name)
     return obj;
 }
 
-Checkpoint *
+CheckpointIn *
 getCheckpoint(const std::string &cpt_dir)
 {
-    return new Checkpoint(cpt_dir, pythonSimObjectResolver);
+    return new CheckpointIn(cpt_dir, pythonSimObjectResolver);
 }
diff --git a/src/python/swig/pyobject.hh b/src/python/swig/pyobject.hh
index a4f06555e..ea2948826 100644
--- a/src/python/swig/pyobject.hh
+++ b/src/python/swig/pyobject.hh
@@ -58,11 +58,11 @@ serializeAll(const std::string &cpt_dir)
     Serializable::serializeAll(cpt_dir);
 }
 
-Checkpoint *
+CheckpointIn *
 getCheckpoint(const std::string &cpt_dir);
 
 inline void
-unserializeGlobals(Checkpoint *cp)
+unserializeGlobals(CheckpointIn &cp)
 {
     Serializable::unserializeGlobals(cp);
 }