LogWrapper takes weight as argument