diff --git a/testcases/cli-test/dim/oe_test_dim_measure_002/oe_test_dim_measure_002.sh b/testcases/cli-test/dim/oe_test_dim_measure_002/oe_test_dim_measure_002.sh
index c297503fc02460d170caa06cfd6efc521369cfdd..31b84ff08da74cf1cca6014a10efc4df88b9e752 100644
--- a/testcases/cli-test/dim/oe_test_dim_measure_002/oe_test_dim_measure_002.sh
+++ b/testcases/cli-test/dim/oe_test_dim_measure_002/oe_test_dim_measure_002.sh
@@ -17,7 +17,7 @@
 source "$OET_PATH/libs/locallibs/common_lib.sh"
 
 function pre_test() {
-    DNF_INSTALL "dim_tools dim"
+    DNF_INSTALL "dim_tools dim gcc"
     modprobe dim_monitor
 }
 
@@ -72,7 +72,7 @@ function run_test() {
 }
 
 function post_test() {
-    DNF_REMOVE
+    DNF_REMOVE "$@"
     pkill -9 dim_proc
     rmmod kvdo dim_monitor dim_core
     rm -rf /etc/dim/