Fix handling of main selection when copying selection to next/previous lines

Fixes #1095

1 files changed, 8 insertions, 2 deletions

diff --git a/src/selection.hh b/src/selection.hh
index 92eecea6..edf8e601 100644
--- a/src/selection.hh
+++ b/src/selection.hh
@@ -101,12 +101,18 @@ struct SelectionList
 
     SelectionList& operator=(Vector<Selection> list)
     {
+        set(std::move(list), list.size()-1);
+        return *this;
+    }
+
+    void set(Vector<Selection> list, size_t main)
+    {
+        kak_assert(main < list.size());
         m_selections = std::move(list);
-        m_main = size()-1;
+        m_main = main;
         sort_and_merge_overlapping();
         update_timestamp();
         check_invariant();
-        return *this;
     }
 
     using iterator = Vector<Selection>::iterator;