diff --git a/SHARP-Track/Display_Probe_Track.m b/SHARP-Track/Display_Probe_Track.m
index 7fc4189..0cf97a8 100644
--- a/SHARP-Track/Display_Probe_Track.m
+++ b/SHARP-Track/Display_Probe_Track.m
@@ -156,7 +156,13 @@
 out_of_brain = false;
 while ~(ann==1 && out_of_brain) % && distance_stepped > .5*active_probe_length)
     m = m-p; % step 10um, backwards up the track
-    ann = av_plot(round(m(1)),round(m(2)),round(m(3))); %until hitting the top
+    if strcmp(plane,'coronal')
+        ann = av_plot(round(m(1)),round(m(2)),round(m(3))); %until hitting the top
+    elseif strcmp(plane,'sagittal')
+        ann = av_plot(round(m(3)),round(m(2)),round(m(1))); %until hitting the top
+    elseif strcmp(plane,'transverse')
+        ann = av_plot(round(m(2)),round(m(3)),round(m(1))); %until hitting the top
+    end
     if strcmp(st.safe_name(ann), 'root')
         % make sure this isn't just a 'root' area within the brain
         m_further_up = m - p*20; % is there more brain 200 microns up along the track?