diff --git a/source/Rbm.cpp b/source/Rbm.cpp
index 83e9bf2..24d8022 100644
--- a/source/Rbm.cpp
+++ b/source/Rbm.cpp
@@ -370,7 +370,7 @@ void Rbm::train(arma::mat const &batch, IListener* pListener)
 	} // number of mini batches
 
 	// Update final status
-	status.err_total = rms_error_accu(batch - prob(h_to_v(prob(v_to_h(batch)))));
+	status.err_total = rms_error_accu(batch - toVisibleProbs(toHiddenProbs(batch)));
 
 	if (pListener)
 	{