diff --git a/src/autoimprove1.py b/src/autoimprove1.py
index af4fb1b..4b997ba 100755
--- a/src/autoimprove1.py
+++ b/src/autoimprove1.py
@@ -171,8 +171,9 @@ def ATest():
 		# check the output
 		print(lines)
 		total_round+=1
-		if lines[0]=='YOU WIN!\n':
-			win_round+=1
+		if len(lines)>0:
+			if lines[0]=='YOU WIN!\n':
+				win_round+=1
 		print("win rate: "+str(win_round/total_round),win_round,total_round)
 	return win_round/total_round