Fixed an error the examples supplying wrong variables when initialising the NN models, due to the change in the previous commit