Bug fixes

2 files changed, 3 insertions, 3 deletions

diff --git a/include/framework/DynamicExtension.h b/include/framework/DynamicExtension.h
index 91bed98..ef36de3 100644
--- a/include/framework/DynamicExtension.h
+++ b/include/framework/DynamicExtension.h
@@ -445,6 +445,7 @@ private:
       /* we first construct a shard from the buffer */
       auto buffview = args->version->get_buffer();
       assert(buffview.get_tail() != buffview.get_head());
+      size_t new_head = buffview.get_tail();
 
       auto new_shard = std::make_shared<ShardType>(std::move(buffview));
 
@@ -471,7 +472,7 @@ private:
         args->version->set_structure(std::move(structure));
       }
 
-      args->version->advance_buffer_head();
+      args->version->advance_buffer_head(new_head);
     }
 
     /* perform all of the reconstructions */
diff --git a/include/framework/scheduling/Version.h b/include/framework/scheduling/Version.h
index 9c62ea1..3e93202 100644
--- a/include/framework/scheduling/Version.h
+++ b/include/framework/scheduling/Version.h
@@ -84,8 +84,7 @@ public:
     return version;
   }
 
-  bool advance_buffer_head() {
-    auto new_head = m_buffer->get_buffer_view().get_tail();
+  bool advance_buffer_head(size_t new_head) {
     m_buffer_head = new_head;
     return m_buffer->advance_head(new_head);
   }