diff --git a/tier_3/data_assimilation/model/neuralnet.py b/tier_3/data_assimilation/model/neuralnet.py
index 3578ac8..79569ae 100644
--- a/tier_3/data_assimilation/model/neuralnet.py
+++ b/tier_3/data_assimilation/model/neuralnet.py
@@ -3,6 +3,7 @@
 import pickle
 from tensorflow.keras.models import load_model
 from .lorenz96 import l96
+import numpy as np
 
 class Periodic1DPadding(Layer):
     """Add a periodic padding to the output of a layer
@@ -53,4 +54,4 @@ def __call__(self, q):
         dq_phi = super().__call__(q)
         #dq_nn =  self.scaler_y.inverse_transform(self.nn.predict(self.scaler_x.transform(q), verbose=0).squeeze())
         dq_nn =  self.scaler_y.inverse_transform(self.nn(self.scaler_x.transform(q)).numpy().squeeze())
-        return dq_phi + dq_nn
\ No newline at end of file
+        return dq_phi + dq_nn