diff --git a/funannotate/library.py b/funannotate/library.py
index 4d0ae5a..137b90f 100755
--- a/funannotate/library.py
+++ b/funannotate/library.py
@@ -11069,7 +11069,7 @@ def trainAugustus(
                 train_table = print_table(trainTable, return_str=True)
                 sys.stderr.write(train_table)
                 # clean up tmp folder
-                shutil.rmtree(trainingdir)
+                SafeRemove(trainingdir)
             else:
                 if train_results[4] < 0.50:
                     log.info(