diff -r c8962b24fb50 tools/python/xen/xend/XendCheckpoint.py
--- a/tools/python/xen/xend/XendCheckpoint.py	Fri Jan 30 11:12:57 2009 +0900
+++ b/tools/python/xen/xend/XendCheckpoint.py	Thu Feb 05 13:58:35 2009 -0700
@@ -332,6 +332,7 @@ def restore(xd, fd, dominfo = None, paus
         if not paused:
             dominfo.unpause()
 
+        xc.tmem_control(0,TMEMC_THAW,-1,0,0,"")
         return dominfo
     except:
         dominfo.destroy()
diff -r c8962b24fb50 tools/python/xen/xend/XendDomainInfo.py
--- a/tools/python/xen/xend/XendDomainInfo.py	Fri Jan 30 11:12:57 2009 +0900
+++ b/tools/python/xen/xend/XendDomainInfo.py	Thu Feb 05 13:58:35 2009 -0700
@@ -2428,6 +2428,7 @@ class XendDomainInfo:
             self.info['start_time'] = time.time()
 
             self._stateSet(DOM_STATE_RUNNING)
+            xc.tmem_control(0,TMEMC_THAW,-1,0,0,"")
         except VmError, exn:
             log.exception("XendDomainInfo.initDomain: exception occurred")
             if self.image:
diff -r c8962b24fb50 tools/python/xen/xend/balloon.py
--- a/tools/python/xen/xend/balloon.py	Fri Jan 30 11:12:57 2009 +0900
+++ b/tools/python/xen/xend/balloon.py	Thu Feb 05 13:58:35 2009 -0700
@@ -26,6 +26,7 @@ from XendLogging import log
 from XendLogging import log
 from XendError import VmError
 import osdep
+from xen.xend.XendConstants import *
 
 RETRY_LIMIT = 20
 RETRY_LIMIT_INCR = 5
@@ -156,6 +157,7 @@ def free(need_mem, dominfo):
             retries = 0
             sleep_time = SLEEP_TIME_GROWTH
 
+        xc.tmem_control(0,TMEMC_FREEZE,-1, 0, 0, "")
         while retries < rlimit:
             physinfo = xc.physinfo()
             free_mem = physinfo['free_memory']