diff --git a/lib/Cli.rb b/lib/Cli.rb
index e7507a4..0e02aa5 100644
--- a/lib/Cli.rb
+++ b/lib/Cli.rb
@@ -49,8 +49,8 @@ class Cli
 	# Display the query's result
 	# @args: search -> the string to search
 	#        protocol -> search from a particular protocol
-	def display(search, protocol=nil, format=nil)
-		result = @m.search(search, protocol)
+	def display(search, protocol=nil, group=nil, format=nil)
+		result = @m.search(search, group, protocol)
 
 		if not result.empty?
 			result.each do |r|
diff --git a/lib/CliSSH.rb b/lib/CliSSH.rb
index bc11593..4f6cd14 100644
--- a/lib/CliSSH.rb
+++ b/lib/CliSSH.rb
@@ -12,7 +12,7 @@ class CliSSH < Cli
 	# Connect to SSH
 	# args: search -> string to search
 	def ssh(search)
-		result = @m.search(search, 'ssh')
+		result = @m.search(search, nil, 'ssh')
 
 		if result.length > 0
 			result.each do |r|