Fixed weights bug