FIX: correct logic

correct logic for key comparison
Merge pull request #4 from dengemann/plot-catch

FIX: correct logic for key errors