diff --git a/toolium/test/utils/test_dataset_map_param_context.py b/toolium/test/utils/test_dataset_map_param_context.py
index eaa665a0..140a3cff 100644
--- a/toolium/test/utils/test_dataset_map_param_context.py
+++ b/toolium/test/utils/test_dataset_map_param_context.py
@@ -152,8 +152,9 @@ class Context(object):
     context = Context()
     context.attribute = "attribute value"
     context.feature_storage = {}
+    context.storage = {"storage_key": "previous storage entry value"}
     dataset.store_key_in_storage(context, "[FEATURE:storage_key]", "feature storage entry value")
-    context.storage = {"storage_key": "storage entry value"}
+    dataset.store_key_in_storage(context, "[storage_key]", "storage entry value")
     dataset.behave_context = context
 
     result_st = map_param("[CONTEXT:storage_key]")