diff --git a/supirfactor_dynamical/_io/_loader.py b/supirfactor_dynamical/_io/_loader.py
index 3854e60..aeebb15 100644
--- a/supirfactor_dynamical/_io/_loader.py
+++ b/supirfactor_dynamical/_io/_loader.py
@@ -113,7 +113,10 @@ def read(
 
     for encoded_arg in _SERIALIZE_ENCODED_ARGS:
         if encoded_arg in kwargs:
-            kwargs[encoded_arg] = _DECODE_ACTIVATIONS[kwargs[encoded_arg]]
+            if kwargs[encoded_arg] is None:
+                pass
+            else:
+                kwargs[encoded_arg] = _DECODE_ACTIVATIONS[kwargs[encoded_arg]]
 
     # Do special loading stuff for the big biophysical model
     if _state_model == 'biophysical':
diff --git a/supirfactor_dynamical/training/train_simple_decoders.py b/supirfactor_dynamical/training/train_simple_decoders.py
index ec7ced1..18222a9 100644
--- a/supirfactor_dynamical/training/train_simple_decoders.py
+++ b/supirfactor_dynamical/training/train_simple_decoders.py
@@ -170,8 +170,7 @@ def train_simple_multidecoder(
                 axis=0,
                 weights=np.array(_validation_n)
             ),
-            validation_n=np.sum(_validation_n),
-
+            validation_n=np.sum(_validation_n)
         )
 
         model_ref.current_epoch = epoch_num