diff --git a/robot/resources/lib/python_keywords/utility_keywords.py b/robot/resources/lib/python_keywords/utility_keywords.py
index 3629dd4..df0072f 100644
--- a/robot/resources/lib/python_keywords/utility_keywords.py
+++ b/robot/resources/lib/python_keywords/utility_keywords.py
@@ -55,7 +55,7 @@ def get_file_hash(filename: str):
 
 @keyword('Generate Wallet')
 def generate_wallet():
-    return wallet.init_wallet_w_addr(ASSETS_DIR)
+    return wallet.init_wallet(ASSETS_DIR)
 
 
 # TODO: should be deleted in the scope