diff --git a/src/ReservoirComputing.jl b/src/ReservoirComputing.jl
index d4dec0da..37863e45 100644
--- a/src/ReservoirComputing.jl
+++ b/src/ReservoirComputing.jl
@@ -19,7 +19,7 @@ export AbstractLayer, create_layer
 export WeightedLayer, DenseLayer, SparseLayer, MinimumLayer, InformedLayer, NullLayer
 export BernoulliSample, IrrationalSample
 export GaussianProcess
-export AbstractReservoir, create_reservoir
+export AbstractReservoir, create_reservoir, create_states
 export RandSparseReservoir, PseudoSVDReservoir, DelayLineReservoir
 export DelayLineBackwardReservoir, SimpleCycleReservoir, CycleJumpsReservoir, NullReservoir
 export RNN, MRNN, GRU, GRUParams, FullyGated, Variant1, Variant2, Variant3, Minimal
diff --git a/src/esn/esn_reservoir_drivers.jl b/src/esn/esn_reservoir_drivers.jl
index e339a3c3..6467df86 100644
--- a/src/esn/esn_reservoir_drivers.jl
+++ b/src/esn/esn_reservoir_drivers.jl
@@ -1,9 +1,12 @@
 abstract type AbstractReservoirDriver end
 
 """
-    create_states(
-        reservoir_driver::AbstractReservoirDriver, train_data, reservoir_matrix,input_matrix
-    )
+    create_states(reservoir_driver::AbstractReservoirDriver,
+                       train_data,
+                       washout,
+                       reservoir_matrix,
+                       input_matrix,
+                       bias_vector)
 
 Return the trained ESN states according to the given driver.
 """