diff --git a/git_cl/PRESUBMIT.py b/git_cl/PRESUBMIT.py
index aaeafaf38..8d2b16050 100644
--- a/git_cl/PRESUBMIT.py
+++ b/git_cl/PRESUBMIT.py
@@ -63,7 +63,7 @@ def RunTests(input_api, output_api):
       proc.communicate()
       if proc.returncode != 0:
         output.append(output_api.PresubmitError('%s failed' % test))
-  except setup_mock.Failure, e:
+  except local_rietveld.Failure, e:
     output.append(output_api.PresubmitError('\n'.join(str(i) for i in e.args)))
   finally:
     server.stop_server()