iₜ=(Wᵢ·xₜ+Uᵢ·hₜ₋₁+Bᵢ)
 fₜ=σ(Wf·xₜ+Uf·hₜ₋₁+Bf)
 oₜ=σ(Wₒ·xₜ+Uₒ·hₜ₋₁+Bₒ)
 ĉₜ=tanh(Wc·xₜ+Uc·hₜ₋₁+Bc)
 cₜ=fₜ*cₜ₋₁+iₜ*ĉₜ
 hₜ=oₜ*tanh(cₜ)
 y=Wy·hₜ+By

diff --git a/train.go b/train.go
index b67f296..f8252aa 100644
--- a/train.go
+++ b/train.go
@@ -121,10 +121,10 @@ func (m *Model) Train(ctx context.Context, dset datasetter.FullTrainer, solver G
                                }
                                copy(hiddenT.Data().([]float32), hidden.Value().Data().([]float32))
                                copy(cellT.Data().([]float32), cell.Value().Data().([]float32))
-                               solver.Step(G.Nodes{
+                               solver.Step(G.NodesToValueGrads(G.Nodes{
                                        lstm.biasC, lstm.biasF, lstm.biasI, lstm.biasO, lstm.biasY,
                                        lstm.uc, lstm.uf, lstm.ui, lstm.uo,
-                                       lstm.wc, lstm.wf, lstm.wi, lstm.wo, lstm.wy})
+                                       lstm.wc, lstm.wf, lstm.wi, lstm.wo, lstm.wy}))
                        }
                }
        }()