1 files changed, 9 insertions, 1 deletions
diff --git a/demoscript/Freeze/library.py b/demoscript/Freeze/library.py
index 9212c18138f..9bed186678a 100755
--- a/demoscript/Freeze/library.py
+++ b/demoscript/Freeze/library.py
@@ -8,7 +8,8 @@
 #
 # **********************************************************************
 
-import pexpect, sys
+import sys
+import demoscript.pexpect as pexpect
 
 def dequote(s):
     cur = 0
@@ -146,4 +147,11 @@ def run(client, server):
     client.sendline('remove')
     #server.expect(['locate could not find', 'locate found.*but it was dead or destroyed'])
     client.expect('current book no longer exists')
+
+    client.sendline('shutdown')
+    if client != server:
+        server.waitTestSuccess()
+    client.sendline('exit')
+    client.waitTestSuccess()
+
     print "ok"