Changes proposed during review

2 files changed, 3 insertions, 3 deletions

diff --git a/lib/gitlab_keys.rb b/lib/gitlab_keys.rb
index 4034a86..d4c4102 100644
--- a/lib/gitlab_keys.rb
+++ b/lib/gitlab_keys.rb
@@ -9,7 +9,7 @@ class GitlabKeys
   attr_accessor :auth_file, :key
 
   def self.command(key_id)
-    raise KeyError.new("Invalid key_id: #{key_id.inspect}") unless /\A[a-z0-9-]+\z/.match(key_id)
+    raise KeyError.new("Invalid key_id: #{key_id.inspect}") unless /\A[a-z0-9-]+\z/ =~ key_id
     "#{ROOT_PATH}/bin/gitlab-shell #{key_id}"
   end
 
diff --git a/spec/gitlab_keys_spec.rb b/spec/gitlab_keys_spec.rb
index a615eac..faf7c5d 100644
--- a/spec/gitlab_keys_spec.rb
+++ b/spec/gitlab_keys_spec.rb
@@ -14,7 +14,7 @@ describe GitlabKeys do
     end
 
     it 'raises KeyError on invalid input' do
-      expect { described_class.command("\nssh-rsa AAA") }.to raise_error(GitlabKeys::KeyError)
+      expect { described_class.command("\nssh-rsa AAA") }.to raise_error(described_class::KeyError)
     end
   end
 
@@ -29,7 +29,7 @@ describe GitlabKeys do
     end
 
     it 'raises KeyError on invalid input' do
-      expect { described_class.key_line('key-741', "ssh-rsa AAA\nssh-rsa AAA") }.to raise_error(GitlabKeys::KeyError)
+      expect { described_class.key_line('key-741', "ssh-rsa AAA\nssh-rsa AAA") }.to raise_error(described_class::KeyError)
     end
   end