slight wallet code edit