diff --git a/src/rbm/train.py b/src/rbm/train.py
index 25cfc2d..fafed93 100644
--- a/src/rbm/train.py
+++ b/src/rbm/train.py
@@ -49,6 +49,9 @@ def cd_binary_binary(entity: Entity, data_pos: Mat):
 
 	# Gibbs	sampling
 	for _ in range(params.num_gibbs_samples-1):
+		if params.do_gibbs_sample_hidden:
+			h_probs_neg = sample(h_probs_neg)
+
 		data_neg = prob(entity.v_given_h(h_probs_neg))
 		h_probs_neg = prob(entity.h_given_v(data_neg))