Fix weight mapping

Ouch!
1 file changed