diff --git a/archinstall/lib/disk/blockdevice.py b/archinstall/lib/disk/blockdevice.py
index e2fcdae5..422f35aa 100644
--- a/archinstall/lib/disk/blockdevice.py
+++ b/archinstall/lib/disk/blockdevice.py
@@ -1,4 +1,6 @@
+import json
 from ..output import log
+from ..general import SysCommand
 
 class BlockDevice:
 	def __init__(self, path, info=None):
diff --git a/archinstall/lib/disk/helpers.py b/archinstall/lib/disk/helpers.py
index d37dfd9b..8b372f73 100644
--- a/archinstall/lib/disk/helpers.py
+++ b/archinstall/lib/disk/helpers.py
@@ -1,5 +1,6 @@
 import re
 import json
+from .blockdevice import BlockDevice
 from ..exceptions import SysCallError
 from ..general import SysCommand
 from ..output import log