diff --git a/tokenise+train.py b/tokenise+train.py
index 4a0f75d..3dc3cc5 100644
--- a/tokenise+train.py
+++ b/tokenise+train.py
@@ -99,7 +99,7 @@ def main() -> int:
 	if output_dir.is_dir():
 		exts = ['.json', '.gz', '.bin']
 		files = [x for x in output_dir.glob('*') if x.suffix in exts]
-		if len(files) == 3:
+		if len(files) == 4:
 			print("Token + model already exists > " + output_dir.name)
 			q = input("Continue? [y/n]")
 			if q != 'y':