add weight fixes