From a6446c913ccaff14040e8c591344420f69916f71 Mon Sep 17 00:00:00 2001
From: Rod Smith <rod.smith@canonical.com>
Date: Thu, 18 Jun 2020 14:06:11 -0400
Subject: Modify import & call to Disk library in fstrim_test script, per
 jocave's comment

---
 bin/fstrim_test | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/bin/fstrim_test b/bin/fstrim_test
index 94556059..c6f1881b 100755
--- a/bin/fstrim_test
+++ b/bin/fstrim_test
@@ -37,14 +37,14 @@ from subprocess import (
     PIPE,
     Popen
 )
-import checkbox_support.disk_support
+from checkbox_support.disk_support import Disk
 
 
 def mount_filesystem(device):
     if "/dev" not in device and device != "":
         device = "/dev/" + device
 
-    test_disk = checkbox_support.disk_support.Disk(device)
+    test_disk = Disk(device)
     mount_point = None
     if test_disk.mount_filesystem(False):
         mount_point = test_disk.get_mount_point()
-- 
cgit v1.2.3