diff --git a/cloudbaseinit/tests/utils/windows/storage/test_factory.py b/cloudbaseinit/tests/utils/windows/storage/test_factory.py
index 987d21db..d189bf46 100644
--- a/cloudbaseinit/tests/utils/windows/storage/test_factory.py
+++ b/cloudbaseinit/tests/utils/windows/storage/test_factory.py
@@ -47,7 +47,7 @@ class TestStorageManager(unittest.TestCase):
             return
 
         self.mock_os.name = "nt"
-        mock_get_os_utils.return_value.check_os_version.return_value = nano
+        mock_get_os_utils.return_value.is_nano_server.return_value = nano
         mock_load_class = mock_class_loader.return_value.load_class
         response = self.factory.get_storage_manager()
         if nano:
diff --git a/cloudbaseinit/utils/windows/storage/factory.py b/cloudbaseinit/utils/windows/storage/factory.py
index a2d176ed..1c33a00a 100644
--- a/cloudbaseinit/utils/windows/storage/factory.py
+++ b/cloudbaseinit/utils/windows/storage/factory.py
@@ -30,7 +30,7 @@ def get_storage_manager():
     cl = classloader.ClassLoader()
 
     if os.name == "nt":
-        if osutils.check_os_version(10, 0):
+        if osutils.is_nano_server():
             # VDS is not available on Nano Server
             # WSM supersedes VDS since Windows Server 2012 / Windows 8
             return cl.load_class(class_paths["WSM"])()