fix simple wallet tests