diff --git a/lib/tests/test_wallet.py b/lib/tests/test_wallet.py
index a7021ef69..62831609e 100644
--- a/lib/tests/test_wallet.py
+++ b/lib/tests/test_wallet.py
@@ -55,8 +55,9 @@ class TestWalletStorage(WalletTestCase):
         storage = WalletStorage(self.wallet_path)
 
         some_dict = {"a":"b", "c":"d"}
-        storage.data = some_dict
 
+        for key, value in some_dict.items():
+            storage.put(key, value, False)
         storage.write()
 
         contents = ""