diff --git a/train.py b/train.py
index 0ab05932..5a1e0cd4 100644
--- a/train.py
+++ b/train.py
@@ -168,7 +168,7 @@ def train(
         model, optimizer = amp.initialize(model, optimizer, opt_level='O1')
 
     # Remove old results
-    for f in glob.glob('train_batch*.jpg') + glob.glob('test_batch*.jpg') + 'results.txt':
+    for f in glob.glob('train_batch*.jpg') + glob.glob('test_batch*.jpg') + ['results.txt']:
         os.remove(f)
 
     # Start training