diff --git a/source/Rbm.cpp b/source/Rbm.cpp
index 5da34b1..ae05cf4 100644
--- a/source/Rbm.cpp
+++ b/source/Rbm.cpp
@@ -131,13 +131,13 @@ void Rbm::gibbs_hv(arma::mat &h_probs, arma::mat &v_probs) const
 void Rbm::contrastiveDivergence(arma::mat const &v_states, arma::mat &dw, arma::mat &dbhv, arma::mat &dbv)
 {
 	arma::mat v_probs(v_states);
-	arma::mat h_states;
+	arma::mat h_states = v_to_h(v_states);
 	arma::mat h_probs;
 	
 	// Sample hidden
-	if (m_params.doGaussianVisible)
+	if (m_params.doGaussianHidden)
 	{
-		h_probs = v_to_h(v_states);
+		h_probs = h_states;
 		h_states = h_probs + arma::randn(h_probs.n_rows, h_probs.n_cols);
 	}
 	else if (m_params.doRaoBlackwell)