1 files changed, 4 insertions, 4 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