diff --git a/pisi/atomicoperations.py b/pisi/atomicoperations.py
index c40c8da0..4ba81d14 100644
--- a/pisi/atomicoperations.py
+++ b/pisi/atomicoperations.py
@@ -188,7 +188,7 @@ class Install(AtomicOperation):
         try:
             int(release)
             pisi.version.make_version(version)
-        except (ValueError, InvalidVersionError):
+        except (ValueError, pisi.version.InvalidVersionError):
             raise Error(_("%s-%s is not a valid PiSi version format") % (version, release))
 
     def check_relations(self):
diff --git a/pisi/operations/build.py b/pisi/operations/build.py
index 09fa90bb..a80260a3 100644
--- a/pisi/operations/build.py
+++ b/pisi/operations/build.py
@@ -589,7 +589,7 @@ class Builder:
         try:
             int(release)
             pisi.version.make_version(version)
-        except (ValueError, InvalidVersionError):
+        except (ValueError, pisi.version.InvalidVersionError):
             raise Error(_("%s-%s is not a valid PiSi version format") % (version, release))
 
     def check_build_dependencies(self):