From 09f0cd3e7130e1a005c40b2f713ec273bdcf269b Mon Sep 17 00:00:00 2001
From: Marcel Paffrath <marcel@minos01.geophysik.ruhr-uni-bochum.de>
Date: Wed, 7 Oct 2015 14:51:00 +0200
Subject: [PATCH] shots no longer None if they are deleted, but flag = 0

---
 pylot/core/active/surveyPlotTools.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pylot/core/active/surveyPlotTools.py b/pylot/core/active/surveyPlotTools.py
index 2f7b2a0a..d7d8318a 100644
--- a/pylot/core/active/surveyPlotTools.py
+++ b/pylot/core/active/surveyPlotTools.py
@@ -160,7 +160,7 @@ class regions(object):
 
     def highlightPicksForShot(self, shot, annotations = False):
         for traceID in shot.getTraceIDlist():
-            if shot.getPick(traceID) is not None:
+            if shot.getFlag(traceID) is not 0:
                 self.highlightPick(shot, traceID, annotations)
         self.refreshFigure()