diff --git a/bsrem_saga.py b/bsrem_saga.py
index a164e62..3d0ad46 100644
--- a/bsrem_saga.py
+++ b/bsrem_saga.py
@@ -44,6 +44,7 @@ def __init__(self, data, initial,
         '''
         super().__init__(**kwargs)
         self.x = initial.copy()
+        self.initial = initial.copy()
         self.data = data
         self.num_subsets = len(data)