lib/gizzmo.rb in gizzmo-0.3.3 vs lib/gizzmo.rb in gizzmo-0.3.4

@@ -196,10 +196,10 @@
     STDERR.puts e.message
     STDERR.puts("=" * 80)
   end
   STDERR.puts subcommands[subcommand_name]
   exit 1
-rescue ThriftClient::Simple::ThriftException => e
+rescue ThriftClient::Simple::ThriftException, Gizzard::Thrift::ShardException => e
   STDERR.puts e.message
   exit 1
 rescue Errno::EPIPE
   # This is just us trying to puts into a closed stdout.  For example, if you pipe into
   # head -1, then this script will keep running after head closes.  We don't care, and
\ No newline at end of file