1 files changed, 4 insertions, 3 deletions
diff --git a/include/framework/reconstruction/LevelingPolicy.h b/include/framework/reconstruction/LevelingPolicy.h
index 6e20cdb..955bc02 100644
--- a/include/framework/reconstruction/LevelingPolicy.h
+++ b/include/framework/reconstruction/LevelingPolicy.h
@@ -21,7 +21,7 @@ class LevelingPolicy : public ReconstructionPolicy<ShardType, QueryType> {
       LevelVector;
 
 public:
-  LevelingPolicy(size_t scale_factor, size_t buffer_size, size_t modifier = 0)
+  LevelingPolicy(size_t scale_factor, size_t buffer_size, double modifier = 0)
       : m_scale_factor(scale_factor), m_buffer_size(buffer_size),
         m_size_modifier(modifier) {}
 
@@ -95,11 +95,12 @@ private:
 
   inline size_t capacity(level_index level, size_t reccnt) const {
     return m_buffer_size *
-           pow(m_scale_factor * pow(std::log(reccnt), m_size_modifier), level);
+           pow(m_scale_factor * std::ceil(std::pow<double>(std::log10(reccnt), m_size_modifier)), level);
+
   }
 
   size_t m_scale_factor;
   size_t m_buffer_size;
-  size_t m_size_modifier;
+  double m_size_modifier;
 };
 } // namespace de