lib/eucalypt/security/namespaces/security-pundit/cli/security-pundit.rb in eucalypt-0.3.1 vs lib/eucalypt/security/namespaces/security-pundit/cli/security-pundit.rb in eucalypt-0.3.2

@@ -1,18 +1,20 @@
 require 'thor'
 require 'eucalypt/helpers'
 require 'eucalypt/security/helpers'
 require 'eucalypt/security/namespaces/security-pundit/generators/role'
+require 'eucalypt/list'
 
 module Eucalypt
   class SecurityPundit < Thor
     include Thor::Actions
     include Eucalypt::Helpers
     include Eucalypt::Helpers::Messages
     include Eucalypt::Helpers::Gemfile
     include Eucalypt::Security::Helpers
     using Colorize
+    extend Eucalypt::List
 
     def self.source_root
       File.join File.dirname(__dir__), 'templates'
     end
 
@@ -66,14 +68,10 @@
       else
         Eucalypt::Error.wrong_directory
       end
     end
 
-    class << self
-      require 'eucalypt/list'
-      include Eucalypt::List
-      def banner(task, namespace = false, subcommand = true)
-        "#{basename} security #{task.formatted_usage(self, true, subcommand).split(':').join(' ')}"
-      end
+    def self.banner(task, namespace = false, subcommand = true)
+      "#{basename} security #{task.formatted_usage(self, true, subcommand).split(':').join(' ')}"
     end
   end
 end
\ No newline at end of file