diff --git a/tests/laser/state/storage_test.py b/tests/laser/state/storage_test.py
index 5fa6078f..a0da2c94 100644
--- a/tests/laser/state/storage_test.py
+++ b/tests/laser/state/storage_test.py
@@ -39,7 +39,7 @@ def test_storage_set_item():
     storage.put(key=1, value=13, addr=10)
 
     # Assert
-    assert storage.put(key=1, value=13, addr=10)
+    assert storage.get(item=1, addr=10) == 13
 
 
 def test_storage_change_item():