fix bugs in RnnWFWBWS