diff --git a/tests/unit/test.py b/tests/unit/test.py
index 558820e3..d7c7b026 100644
--- a/tests/unit/test.py
+++ b/tests/unit/test.py
@@ -22,8 +22,8 @@ from oslotest import base
 from oslotest import mockpatch
 
 from rally.common import db
-from rally.common import objects
 from rally import plugins
+from rally.task import utils as tutils
 from tests.unit import fakes
 
 
@@ -47,9 +47,8 @@ class TestCase(base.BaseTestCase):
         plugins.load()
 
     def _test_atomic_action_timer(self, atomic_actions, name):
-        _old_atomic_actions = objects.Task.convert_atomic_actions(
-            atomic_actions)
-        action_duration = _old_atomic_actions.get(name)
+        atomic_wrapper = tutils.WrapperForAtomicActions(atomic_actions)
+        action_duration = atomic_wrapper.get(name)
         self.assertIsNotNone(action_duration)
         self.assertIsInstance(action_duration, float)