1 files changed, 8 insertions, 1 deletions
diff --git a/demo_windows.py b/demo_windows.py
index 5d870907..84e3ec60 100755
--- a/demo_windows.py
+++ b/demo_windows.py
@@ -9,11 +9,18 @@
 import sys, os, base64, getpass, socket, traceback
 import paramiko
 
+if os.environ.has_key('HOME'):
+    # unix
+    HOME = os.environ['HOME']
+else:
+    # windows
+    HOME = os.environ['HOMEDRIVE'] + os.environ['HOMEPATH']
+
 
 #####   utility functions
 
 def load_host_keys():
-    filename = os.environ['HOME'] + '/.ssh/known_hosts'
+    filename = HOME + '/.ssh/known_hosts'
     keys = {}
     try:
         f = open(filename, 'r')