diff --git a/testcases/cli-test/gdisk/oe_test_gdisk_create_partition_128.sh b/testcases/cli-test/gdisk/oe_test_gdisk_create_partition_128.sh
index ca07d5f18a41405f9126134e7d401d346dedc5fc..78e91e268cd8221c7420d57c8249181bdf0f9e7b 100755
--- a/testcases/cli-test/gdisk/oe_test_gdisk_create_partition_128.sh
+++ b/testcases/cli-test/gdisk/oe_test_gdisk_create_partition_128.sh
@@ -21,7 +21,7 @@ source "${OET_PATH}"/libs/locallibs/common_lib.sh
 source "${OET_PATH}"/testcases/cli-test/gdisk/common_lib.sh
 
 function pre_test() {
-	DNF_INSTALL gdisk
+	DNF_INSTALL "gdisk parted"
 	get_unused_disk
 	[ -z "$unused_disk" ] && echo "no unused disk" && exit 1
 }
@@ -29,6 +29,7 @@ function pre_test() {
 function run_test() {
 	create_partition 128 10M
 	partprobe
+	lsblk
 	num=$(lsblk | grep -c "${unused_disk#/dev/}")
 	CHECK_RESULT "$num" 129
 	for i in $(seq 1 128); do
@@ -55,7 +56,7 @@ function run_test() {
 function post_test() {
 	rm -rf mount_tmp*
 	partprobe
-	DNF_REMOVE
+	DNF_REMOVE "$@"
 }
 
 main "$@"