small bug in CrossRNN update part