diff --git a/disk.py b/disk.py
index bfa11a1..43d8124 100644
--- a/disk.py
+++ b/disk.py
@@ -1,3 +1,5 @@
+import shutil
+
 import json
 import shutil