diff --git a/shared_module/container/detail/small_vector_base.hpp b/shared_module/container/detail/small_vector_base.hpp
index c90a172..acafc9d 100644
--- a/shared_module/container/detail/small_vector_base.hpp
+++ b/shared_module/container/detail/small_vector_base.hpp
@@ -245,7 +245,7 @@ public:
     {
         if (count <= N) {
             if (size() > N) {
-                m_stack.assign(std::make_move_iterator(m_heap->begin()), std::make_move_iterator(m_heap->end()));
+                m_stack.assign(std::make_move_iterator(m_heap->begin()), std::make_move_iterator(m_heap->begin() + count));
                 m_heap->clear();
                 m_first = m_stack.begin();
             } else {
@@ -263,7 +263,7 @@ public:
     {
         if (count <= N) {
             if (size() > N) {
-                m_stack.assign(std::make_move_iterator(m_heap->begin()), std::make_move_iterator(m_heap->end()));
+                m_stack.assign(std::make_move_iterator(m_heap->begin()), std::make_move_iterator(m_heap->begin() + count));
                 m_heap->clear();
                 m_first = m_stack.begin();
             } else {