diff --git a/source/Rbm.cpp b/source/Rbm.cpp
index 2a73a68..ae6fc7c 100644
--- a/source/Rbm.cpp
+++ b/source/Rbm.cpp
@@ -14,7 +14,7 @@
 #include <cassert>
 #include "Rbm.hpp"
 
-#define RBM_TRAIN_FLAT 1
+#define RBM_TRAIN_FLAT 0
 
 Rbm::Rbm(size_t numVisible, size_t numHidden, size_t numContext)
 : m_params()