diff --git a/test/functional/api/cas/cas_module.py b/test/functional/api/cas/cas_module.py
index 09faa7e..58f0dd6 100644
--- a/test/functional/api/cas/cas_module.py
+++ b/test/functional/api/cas/cas_module.py
@@ -2,7 +2,9 @@
 # Copyright(c) 2019 Intel Corporation
 # SPDX-License-Identifier: BSD-3-Clause-Clear
 #
+
 from aenum import Enum
+
 from test_utils import os_utils
 from test_utils.os_utils import ModuleRemoveMethod
 
@@ -13,5 +15,12 @@ class CasModule(Enum):
 
 
 def reload_all_cas_modules():
-    os_utils.unload_kernel_module(CasModule.cache, ModuleRemoveMethod.modprobe)
-    os_utils.load_kernel_module(CasModule.cache)
+    os_utils.unload_kernel_module(CasModule.cache.value, ModuleRemoveMethod.modprobe)
+    os_utils.load_kernel_module(CasModule.cache.value)
+
+
+def unload_all_cas_modules():
+    os_utils.unload_kernel_module(CasModule.cache.value,
+                                  os_utils.ModuleRemoveMethod.rmmod)
+    os_utils.unload_kernel_module(CasModule.disk.value,
+                                  os_utils.ModuleRemoveMethod.rmmod)