Added missing weight update call.