2 files changed, 3 insertions, 3 deletions
diff --git a/src/brain/neural_link.cpp b/src/brain/neural_link.cpp
index c61feb4..6997bd6 100644
--- a/src/brain/neural_link.cpp
+++ b/src/brain/neural_link.cpp
@@ -1,11 +1,11 @@
 #include "brain/neural_link.h"
 
-NeuralLink::NeuralLink(std::weak_ptr<NeuralNode> input, double weight)
+NeuralLink::NeuralLink(NeuralNode* input, double weight)
     :_input(input), _weight(weight)
 {
 }
 
 double NeuralLink::weightedActivation() const
 {
-    return _input.lock()->activation()*_weight;
+    return _input->activation()*_weight;
 }
diff --git a/src/brain/neural_network.cpp b/src/brain/neural_network.cpp
index 64d6979..bacdbed 100644
--- a/src/brain/neural_network.cpp
+++ b/src/brain/neural_network.cpp
@@ -84,7 +84,7 @@ void NeuralNetwork::parseFile(std::istream &&file)
 
 void NeuralNetwork::addLink(std::shared_ptr<NeuralNode> source, std::shared_ptr<Neuron> destination, double weight)
 {
-    NeuralLink link(source, weight);
+    NeuralLink link(source.get(), weight);
     destination->addInput(std::move(link));
 }