diff --git a/src/DLB/LB_Model.hpp b/src/DLB/LB_Model.hpp
index d8f65aaf64e7c4031098088e11203d7fe7100b2a..fabfab4a6690286e8e5f4bd1646c41a9a65568ab 100644
--- a/src/DLB/LB_Model.hpp
+++ b/src/DLB/LB_Model.hpp
@@ -50,7 +50,7 @@ struct ModelSquare
 
 	template<typename Decomposition, typename vector> inline void addComputation(Decomposition & dec, const vector & vd, size_t v, size_t p)
 	{
-		dec.addComputationCost(v, 1);
+		dec.addComputationCost(v, factor);
 	}
 
 	template<typename Decomposition> inline void applyModel(Decomposition & dec, size_t v)
diff --git a/src/Vector/vector_dist_dlb_test.hpp b/src/Vector/vector_dist_dlb_test.hpp
index 27c46d9b356ff8a2fc883c0579fd91808968b241..8abed0d8030c1ac93e139f18180b9b02dd41293c 100644
--- a/src/Vector/vector_dist_dlb_test.hpp
+++ b/src/Vector/vector_dist_dlb_test.hpp
@@ -43,6 +43,7 @@ BOOST_AUTO_TEST_CASE( vector_dist_dlb_test_part )
 	vd.ghost_get<>();
 
 	ModelSquare md;
+	md.factor = 10;
 	vd.addComputationCosts(md);
 	vd.getDecomposition().decompose();
 	vd.map();