1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/Optimizer.cpp b/src/armnn/Optimizer.cpp
index 5e50c01c09..4d0aae8491 100644
--- a/src/armnn/Optimizer.cpp
+++ b/src/armnn/Optimizer.cpp
@@ -32,7 +32,9 @@ void Optimizer::Pass(Graph& graph, const Optimizations& optimizations)
 
             if ((*it)->IsOutputUnconnected())
             {
-                it = graph.EraseLayer(it);
+                auto next = std::next(graph.GetPosInGraph(**it));
+                graph.EraseLayer(it);
+                it = next;
                 graphNeedsSorting = true;
             }