Modify import & call to Disk library in fstrim_test script, per jocave's comment

1 files changed, 2 insertions, 2 deletions

diff --git a/bin/fstrim_test b/bin/fstrim_test
index 9455605..c6f1881 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()