From eb3c11646e686e083ee67f33a5268a32ec7f1519 Mon Sep 17 00:00:00 2001
From: Mark Bakker <markbak@gmail.com>
Date: Thu, 1 Nov 2018 13:59:15 +0100
Subject: [PATCH] initialize line-sinks in initialize function

---
 ttim/linesink.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/ttim/linesink.py b/ttim/linesink.py
index a1e9504..cc72403 100644
--- a/ttim/linesink.py
+++ b/ttim/linesink.py
@@ -374,15 +374,19 @@ def __init__(self, model, xy=[(-1, 0), (1, 0)], tsandh=[(0, 1)], \
         self.x = xy[:, 0]
         self.y = xy[:, 1]
         self.nls = len(self.x) - 1
-        for i in range(self.nls):
-            self.lslist.append(HeadLineSink(model, x1=self.x[i], y1=self.y[i], \
-                                            x2=self.x[i + 1], y2=self.y[i + 1], \
-                                            tsandh=tsandh, res=res, wh=wh, \
-                                            layers=layers, label=None, \
-                                            addtomodel=False) )
+        self.tsandh = tsandh
+        self.res = np.atleast_1d(float(res))
+        self.wh = wh
         self.model.addelement(self)
 
     def initialize(self):
+        self.lslist = []
+        for i in range(self.nls):
+            self.lslist.append(HeadLineSink(self.model, x1=self.x[i], y1=self.y[i], \
+                                            x2=self.x[i + 1], y2=self.y[i + 1], \
+                                            tsandh=self.tsandh, res=self.res, wh=self.wh, \
+                                            layers=self.layers, label=None, \
+                                            addtomodel=False) )
         LineSinkStringBase.initialize(self)
         self.pc = np.zeros(self.nls * self.nlayers)
         for i in range(self.nls):