diff --git a/source/poet.cpp b/source/poet.cpp
index 170bcdc..51c2e30 100644
--- a/source/poet.cpp
+++ b/source/poet.cpp
@@ -88,6 +88,10 @@ int main(int argc, char *argv[])
 	
 	if (command == Command::Train)
 	{
+		for (int i=0; i < stack->numLayers(); i++)
+		{
+			stack->getLayer(i)->params().gibbsDoSampleHidden = true;
+		}
 		RbmListener listener;
 
 		// Phase 10