diff --git a/source/Rbm.cpp b/source/Rbm.cpp
index c6da7f8..ff039c5 100644
--- a/source/Rbm.cpp
+++ b/source/Rbm.cpp
@@ -76,10 +76,10 @@ arma::mat Rbm::toVisibleProbs(const arma::mat& hidden) const
 	return Rbm::prob(h_to_v(hidden));
 }
 
-void Rbm::weightsAssign(const arma::mat& w, const arma::mat& bhv, const arma::mat& bv)
+void Rbm::weightsAssign(const arma::mat& w, const arma::mat& bh, const arma::mat& bv)
 {
 	m_whv.submat(0, 0, w.n_rows - 1, w.n_cols - 1) = w;
-	m_bh.submat(0, 0, bhv.n_rows - 1, bhv.n_cols - 1) = bhv;
+	m_bh.submat(0, 0, bh.n_rows - 1, bh.n_cols - 1) = bh;
 	m_bv.submat(0, 0, bv.n_rows - 1, bv.n_cols - 1) = bv;
 }