diff --git a/qubesmanager/backup.py b/qubesmanager/backup.py
index 5cb6fab..1492fbd 100644
--- a/qubesmanager/backup.py
+++ b/qubesmanager/backup.py
@@ -23,7 +23,6 @@
 import signal
 from qubesadmin import exc
 from qubesadmin import utils as admin_utils
-from qubes.storage.file import get_disk_usage
 
 from PyQt5 import QtCore, QtWidgets  # pylint: disable=import-error
 from . import ui_backupdlg  # pylint: disable=no-name-in-module
@@ -35,6 +34,7 @@ from . import utils
 import grp
 import pwd
 import os
+import shutil
 
 
 # pylint: disable=too-few-public-methods
@@ -202,7 +202,7 @@ class BackupVMsWindow(ui_backupdlg.Ui_Backup, QtWidgets.QWizard):
             if vm.qid == 0:
                 local_user = grp.getgrnam('qubes').gr_mem[0]
                 home_dir = pwd.getpwnam(local_user).pw_dir
-                self.size = get_disk_usage(home_dir)
+                self.size = shutil.disk_usage(home_dir)[1]
             else:
                 self.size = vm.get_disk_utilization()
             super(BackupVMsWindow.VmListItem, self).__init__(
diff --git a/qubesmanager/restore.py b/qubesmanager/restore.py
index 8f873c5..edc4baa 100644
--- a/qubesmanager/restore.py
+++ b/qubesmanager/restore.py
@@ -26,8 +26,6 @@ import os.path
 import logging
 import logging.handlers
 
-from qubes import backup
-
 from . import ui_restoredlg  # pylint: disable=no-name-in-module
 from . import multiselectwidget
 from . import backup_utils
@@ -53,7 +51,7 @@ class RestoreThread(QtCore.QThread):
         try:
             self.backup_restore.restore_do(self.vms_to_restore)
 
-        except backup.BackupCanceledError as ex:
+        except exc.BackupCancelledError as ex:
             self.canceled = True
             err_msg.append(str(ex))
         except Exception as ex:  # pylint: disable=broad-except
diff --git a/test-packages/qubes/backup/__init__.py b/test-packages/qubes/backup/__init__.py
deleted file mode 100644
index dc24fbb..0000000
--- a/test-packages/qubes/backup/__init__.py
+++ /dev/null
@@ -1,5 +0,0 @@
-class BackupCanceledError(BaseException):
-
-    tmpdir = None
-
-    pass
diff --git a/test-packages/qubes/storage/__init__.py b/test-packages/qubes/storage/__init__.py
deleted file mode 100644
index 2ae2839..0000000
--- a/test-packages/qubes/storage/__init__.py
+++ /dev/null
@@ -1 +0,0 @@
-pass
diff --git a/test-packages/qubes/storage/file.py b/test-packages/qubes/storage/file.py
deleted file mode 100644
index 1503cca..0000000
--- a/test-packages/qubes/storage/file.py
+++ /dev/null
@@ -1,2 +0,0 @@
-def get_disk_usage(*args):
-    return 0
diff --git a/test-packages/qubesadmin/exc.py b/test-packages/qubesadmin/exc.py
index 83310ab..8025cb2 100644
--- a/test-packages/qubesadmin/exc.py
+++ b/test-packages/qubesadmin/exc.py
@@ -13,3 +13,6 @@ class QubesPropertyAccessError(BaseException):
 
 class QubesDaemonNoResponseError(BaseException):
     pass
+
+class BackupCancelledError(BaseException):
+    pass