Revert "small correction on weight"

This reverts commit 3abfb17acde95d77909534e99691c39f7656b24b.