diff --git a/core/module.rb b/core/module.rb
index 8a29c3468..2c16980e4 100644
--- a/core/module.rb
+++ b/core/module.rb
@@ -30,10 +30,10 @@ module Module
         return false
     end
     
-    # Return module configuration by database id
+    # Return module key by database id
     def self.get_key_by_database_id(id)
         ret = BeEF::Core::Configuration.instance.get('beef.module').select {|k, v| v.has_key?('db') and v['db']['id'].to_i == id.to_i }
-        return (ret.kind_of?(Array)) ? ret.first : ret.keys[0]
+        return (ret.kind_of?(Array)) ? ret.first.first : ret.keys.first
     end
 
     # Returns category name in a system folder format