lib/travis/cli/restart.rb in travis-1.5.2 vs lib/travis/cli/restart.rb in travis-1.5.3

@@ -1,12 +1,15 @@
 require 'travis/cli'
 
 module Travis
   module CLI
     class Restart < RepoCommand
+      description "restarts a build or job"
+
       def run(number = last_build.number)
         authenticate
         entity = job(number) || build(number)
+        error "could not find job or build #{repository.slug}##{number}" unless entity
         entity.restart
 
         say "restarted", "#{entity.class.one} ##{entity.number} has been %s"
       end
     end