diff --git a/xfel_calibrate/finalize.py b/xfel_calibrate/finalize.py
index cdba7b8b8647d85fff6d54061c9b218c8017c820..4a912bc61e9742b7ca256b62bb13007bb7363ff5 100644
--- a/xfel_calibrate/finalize.py
+++ b/xfel_calibrate/finalize.py
@@ -3,7 +3,7 @@ from importlib.machinery import SourceFileLoader
 from os import chdir, listdir, makedirs, path, remove, stat
 from os.path import isdir, isfile, splitext
 import re
-from shutil import copy, copytree, move
+from shutil import copy, copytree, move, rmtree
 from subprocess import CalledProcessError, check_call, check_output
 from textwrap import dedent
 from time import sleep