diff --git a/tokenise+train.py b/tokenise+train.py
index 8b2defb..a5dd7d7 100644
--- a/tokenise+train.py
+++ b/tokenise+train.py
@@ -128,7 +128,7 @@ def main() -> int:
 		output_dir.mkdir()
 
 	encode(filepath=args.text, blocksize=args.blocksize, vocabsize=args.vocabsize, ouputdir=output_dir, linebyline=args.line_by_line)
-	train(ouputdir=output_dir, blocksize=args.blocksize, vocabsize=args.vocabsize, num_steps=args.numsteps, gpu=args.gpu)
+	train(filepath=args.text, ouputdir=output_dir, blocksize=args.blocksize, vocabsize=args.vocabsize, num_steps=args.numsteps, gpu=args.gpu)
 
 
 if __name__ == '__main__':