2 files changed, 5 insertions, 5 deletions
diff --git a/lib/dbshell/client/database.rb b/lib/dbshell/client/database.rb
index 35cb9ee..b39581f 100644
--- a/lib/dbshell/client/database.rb
+++ b/lib/dbshell/client/database.rb
@@ -11,7 +11,7 @@ module DBShell
         when 'postgresql'
           DBShell::Client::Postgres
         else
-          raise DBShell::InvalidDatabaseAdapter
+          raise InvalidDatabaseAdapter
         end
       end
 
@@ -21,8 +21,8 @@ module DBShell
           .runshell(connection_params)
       end
     end
-  end
-end
 
 
-class DBShell::InvalidDatabaseAdapter < StandardError; end
+    class InvalidDatabaseAdapter < StandardError; end
+  end
+end
diff --git a/spec/dbshell/client/database_spec.rb b/spec/dbshell/client/database_spec.rb
index 5fe5597..1328cf7 100644
--- a/spec/dbshell/client/database_spec.rb
+++ b/spec/dbshell/client/database_spec.rb
@@ -20,7 +20,7 @@ describe DBShell::Client::Database do
         DBShell::Client::Database.handler({
           'adapter' => 'sybil_system'
         })
-      end.must_raise(DBShell::InvalidDatabaseAdapter)
+      end.must_raise(DBShell::Client::InvalidDatabaseAdapter)
     end
   end
 end