diff --git a/Source/Rbm.hpp b/Source/Rbm.hpp
index a42afe1..dd25ee1 100644
--- a/Source/Rbm.hpp
+++ b/Source/Rbm.hpp
@@ -212,6 +212,7 @@ public:
 				m /= i;
 				th.states().array() = m.array() - m_sparsity;
 				hiddenBiasUpdate(th, -mu);
+				w = m_w;
 
 //				cout << "Mean(" << m_sparsity << ") = " << (double)m.array().mean() << endl;
 //				cout << m << endl;