Fixed weight mapping

Change-Id: Iff019aff3818495206b363d2d00e993507f9cea5
Signed-off-by: rarkenin <rarkenin@gmail.com>