Chos fix to rnn fprop