diff --git a/neural_filters/neural_filter.py b/neural_filters/neural_filter.py
index 4ee01679b293be73f5bf902c9ed810e4c124b247..e80b6c7fe3bac3d7fd15904688cd5d3674b82bb9 100644
--- a/neural_filters/neural_filter.py
+++ b/neural_filters/neural_filter.py
@@ -34,7 +34,7 @@ class NeuralFilter(torch.nn.Module):
 
         self.hidden_size = hidden_size
 
-        self.register_parameter('bias_forget', torch.Tensor(hidden_size))
+        self.register_parameter('bias_forget', Parameter(torch.Tensor(hidden_size)))
 
         self.reset_parameters()
 
diff --git a/neural_filters/neural_filter_2CC.py b/neural_filters/neural_filter_2CC.py
index 03f49ab36db1201614a3b979c5be63c708e14aa2..efbb4aa6e3f81fc83ad95d6987bf1b362b9a6364 100644
--- a/neural_filters/neural_filter_2CC.py
+++ b/neural_filters/neural_filter_2CC.py
@@ -35,8 +35,8 @@ class NeuralFilter2CC(torch.nn.Module):
 
         self.hidden_size = hidden_size
 
-        self.register_parameter('bias_theta', torch.Tensor(hidden_size))
-        self.register_parameter('bias_modulus', torch.Tensor(hidden_size))
+        self.register_parameter('bias_theta', Parameter(torch.Tensor(hidden_size)))
+        self.register_parameter('bias_modulus', Parameter(torch.Tensor(hidden_size)))
 
         self.reset_parameters()