except InvalidAtom in atom.py

1 files changed, 2 insertions, 1 deletions

diff --git a/scripts/atom.py b/scripts/atom.py
index 2e8ca7f..a93251a 100755
--- a/scripts/atom.py
+++ b/scripts/atom.py
@@ -2,12 +2,13 @@
 
 import sys
 from portage.dep import Atom
+from portage.exception import InvalidAtom
 
 for line in sys.stdin.buffer:
     try:
         Atom(line.decode().strip())
         sys.stdout.buffer.write(b"0\n")
-    except:
+    except InvalidAtom:
         sys.stdout.buffer.write(b"1\n")
     finally:
         sys.stdout.buffer.flush()