diff --git a/lib/plugins/inspec-habitat/lib/inspec-habitat/cli.rb b/lib/plugins/inspec-habitat/lib/inspec-habitat/cli.rb index 07d163ce2..a9f8b663c 100644 --- a/lib/plugins/inspec-habitat/lib/inspec-habitat/cli.rb +++ b/lib/plugins/inspec-habitat/lib/inspec-habitat/cli.rb @@ -4,8 +4,6 @@ require 'inspec/dist' module InspecPlugins module Habitat class ProfileCLI < Inspec.plugin(2, :cli_command) - include Inspec::Dist - # Override banner method to correct missing subcommand. # @see https://github.com/erikhuda/thor/issues/261 def self.banner(command, _namespace = nil, _subcommand = false) @@ -31,6 +29,8 @@ module InspecPlugins end class CLI < Inspec.plugin(2, :cli_command) + include Inspec::Dist + subcommand_desc 'habitat SUBCOMMAND', "Manage Habitat with #{PRODUCT_NAME}" namespace 'habitat'