diff --git a/test/functional/tests/misc/test_weak_modules.py b/test/functional/tests/misc/test_weak_modules.py
index 2aadf7c..c4b5c13 100644
--- a/test/functional/tests/misc/test_weak_modules.py
+++ b/test/functional/tests/misc/test_weak_modules.py
@@ -15,7 +15,7 @@ from test_tools.fs_utils import (
     readlink,
     remove,
 )
-from test_tools.packaging import Packages
+from api.cas.packaging import Packages
 
 
 modules_links_dir = "/lib/modules/$(uname -r)/weak-updates/block/opencas"
@@ -54,7 +54,7 @@ def test_weak_modules():
         cas_pkg.create(TestRun.usr.working_dir)
 
     with TestRun.step("Remove any previous installations and cleanup"):
-        cas_pkg.uninstall("open-cas-linux")
+        cas_pkg.uninstall_all_matching("open-cas-linux")
         remove(modules_links_dir, recursive=True, force=True)
 
     with TestRun.step("Install RPM packages and check for module symlinks"):