diff --git a/src/main.cpp b/src/main.cpp
index 01b30ed..95c5a81 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -114,8 +114,7 @@ static void updateVisualisers()
     for (LayerData &layer : *rendererData.currentData) {
         LayerVisualisation* visualisation = getVisualisationForLayer(layer);
         if (prevState && prevVisualisation && visualisation) {
-            auto interaction = getActivityAnimation(*prevState, rendererData.layerInfo.at(layer.name()),
-                                                    prevVisualisation->nodePositions(), visualisation->nodePositions());
+            auto interaction = getActivityAnimation(*prevState, layer, rendererData.layerInfo.at(layer.name()), prevVisualisation->nodePositions(), visualisation->nodePositions());
             rendererData.animations.emplace_back(interaction);
         }
 
diff --git a/src/visualisations.cpp b/src/visualisations.cpp
index 156fcff..43a970a 100644
--- a/src/visualisations.cpp
+++ b/src/visualisations.cpp
@@ -62,8 +62,9 @@ static Animation *getFullyConnectedAnimation(const fmri::LayerData &prevState, c
     return new ActivityAnimation(result, prevPositions.data(), curPositions.data(), -10);
 }
 
-Animation *fmri::getActivityAnimation(const fmri::LayerData &prevState, const fmri::LayerInfo &layer,
-                                      const vector<float> &prevPositions, const vector<float> &curPositions)
+Animation * fmri::getActivityAnimation(const fmri::LayerData &prevState, const fmri::LayerData &curState,
+                                       const fmri::LayerInfo &layer, const vector<float> &prevPositions,
+                                       const vector<float> &curPositions)
 {
     if (prevPositions.empty() || curPositions.empty()) {
         // Not all positions know, no visualisation possible.
diff --git a/src/visualisations.hpp b/src/visualisations.hpp
index 913be57..43549e4 100644
--- a/src/visualisations.hpp
+++ b/src/visualisations.hpp
@@ -14,6 +14,7 @@ namespace fmri {
      */
     LayerVisualisation* getVisualisationForLayer(const LayerData& layer);
 
-    Animation *getActivityAnimation(const fmri::LayerData &prevState, const fmri::LayerInfo &layer,
-                                        const vector<float> &prevPositions, const vector<float> &curPositions);
+    Animation * getActivityAnimation(const fmri::LayerData &prevState, const fmri::LayerData &curState,
+                                     const fmri::LayerInfo &layer, const vector<float> &prevPositions,
+                                     const vector<float> &curPositions);
 }