diff --git a/plotly/plotlyfig_aux/helpers/findLegendAxis.m b/plotly/plotlyfig_aux/helpers/findLegendAxis.m
index a5ea4962..11a67d7b 100644
--- a/plotly/plotlyfig_aux/helpers/findLegendAxis.m
+++ b/plotly/plotlyfig_aux/helpers/findLegendAxis.m
@@ -1,4 +1,9 @@
 function legendAxis = findLegendAxis(obj,legendHandle)
-legendAxisIndex = find(arrayfun(@(x)(isequal(handle(getappdata(x.Handle,'LegendPeerHandle')),legendHandle)),obj.State.Axis), 1);
-legendAxis = obj.State.Axis(legendAxisIndex).Handle; 
-end
\ No newline at end of file
+  if verLessThan('matlab','9.0.0')
+    legendAxisIndex = find(arrayfun(@(x)(isequal(handle(getappdata(x.Handle,'LegendPeerHandle')),legendHandle)),obj.State.Axis), 1);
+  else
+    legendAxisIndex = find(arrayfun(@(x)(isequal(handle(get(x.Handle,'Legend')),legendHandle)),obj.State.Axis), 1);
+  end
+
+  legendAxis = obj.State.Axis(legendAxisIndex).Handle;
+end