diff --git a/tests/unit_tests/pyanaconda_tests/modules/storage/test_module_dasd.py b/tests/unit_tests/pyanaconda_tests/modules/storage/test_module_dasd.py
index ed5c40ce9e9c..4a6f93d2caf0 100644
--- a/tests/unit_tests/pyanaconda_tests/modules/storage/test_module_dasd.py
+++ b/tests/unit_tests/pyanaconda_tests/modules/storage/test_module_dasd.py
@@ -124,13 +124,13 @@ def test_discovery_fails(self):
             DASDDiscoverTask("x.y.z").run()
 
     @patch('pyanaconda.modules.storage.dasd.discover.blockdev')
-    def test_discovery(self, blockdev):
+    @patch('pyanaconda.modules.storage.dasd.discover.execWithRedirect')
+    def test_discovery(self, blockdev, execWithRedirect):
         """Test the discovery task."""
         DASDDiscoverTask("0.0.A100").run()
         blockdev.s390.sanitize_dev_input.assert_called_once_with("0.0.A100")
 
-        sanitized_input = blockdev.s390.sanitize_dev_input.return_value
-        blockdev.s390.dasd_online.assert_called_once_with(sanitized_input)
+        execWithRedirect.assert_called_once()
 
     @patch('pyanaconda.modules.storage.dasd.format.blockdev')
     def test_format(self, blockdev):