1 files changed, 4 insertions, 4 deletions

diff --git a/midi-dump.hs b/midi-dump.hs
index cca4e5c..0082987 100644
--- a/midi-dump.hs
+++ b/midi-dump.hs
@@ -58,7 +58,7 @@ data TriadType = Major | Minor deriving (Show, Eq, Ord)
 data Triad =
        Triad
          { _triadType :: TriadType
-         , triadPitch :: Event.Pitch
+         , _triadPitch :: Event.Pitch
          , _triadVelocity :: Event.Velocity
          }
   deriving (Show, Eq, Ord)
@@ -442,12 +442,12 @@ processMidi = do
         when (Map.null newKeys) $
              modify $ \s -> s { _triadRecording = Recording pitch ttype, _triadMap = Map.delete (pc, ttype) (_triadMap s) }
       Recording pitch ttype -> do
+        triadMap <- gets _triadMap
         let f :: Map (Event.Channel, Event.Pitch) Event.Velocity -> Set ScaleDegree
             f = Set.fromList . map (subtract (fromIntegral $ Event.unPitch pitch) . fromIntegral . Event.unPitch . snd) . Map.keys
             pc = toPitchClass pitch
-        let detected = triadPitch . snd <$> detectTriads newKeys
-            detected :: [Event.Pitch]
-        if pitch `elem` detected then do
+            done = Map.null newKeys && not (Map.null triadMap)
+        if done then do
           modify $ \s -> s { _triadRecording = TriadNotRecording }
           liftIO $ putStrLn "Recorded triad"
         else