2c5bb06
diff --git a/setup.py b/setup.py
2c5bb06
index 58a0ae1..f61ef8a 100755
2c5bb06
--- a/setup.py
2c5bb06
+++ b/setup.py
2c5bb06
@@ -37,7 +37,7 @@ setup(name='dropbox',
2c5bb06
       author='Dropbox, Inc.',
2c5bb06
       author_email='support-api@dropbox.com',
2c5bb06
       url='http://www.dropbox.com/',
2c5bb06
-      packages=['dropbox', 'tests'],
2c5bb06
+      packages=['dropbox'],
2c5bb06
       install_requires=INSTALL_REQUIRES,
2c5bb06
       package_data={'dropbox': ['trusted-certs.crt'],
2c5bb06
                     'tests' : ['server.crt', 'server.key']},
2c5bb06
diff --git a/tests/test_client.py b/tests/test_client.py
2c5bb06
index 87b0b0b..b74c31d 100755
2c5bb06
--- a/tests/test_client.py
2c5bb06
+++ b/tests/test_client.py
2c5bb06
@@ -292,7 +292,7 @@ class TestClientOAuth1(BaseClientTests, unittest.TestCase):
2c5bb06
         sess.set_token(cls.access_key, cls.access_secret)
2c5bb06
 
2c5bb06
 class TestClientOAuth2(BaseClientTests, unittest.TestCase):
2c5bb06
-    Session = session.DropboxSessionOAuth2
2c5bb06
+    Session = session.DropboxOAuth2Session
2c5bb06
 
2c5bb06
     @classmethod
2c5bb06
     def _configure_session(cls, sess):
2c5bb06
diff --git a/tests/test_session.py b/tests/test_session.py
2c5bb06
index bca077a..f0b34f6 100644
2c5bb06
--- a/tests/test_session.py
2c5bb06
+++ b/tests/test_session.py
2c5bb06
@@ -3,7 +3,7 @@ import unittest
2c5bb06
 import urllib
2c5bb06
 import urlparse
2c5bb06
 
2c5bb06
-from dropbox.session import DropboxSession, DropboxSessionOAuth2
2c5bb06
+from dropbox.session import DropboxSession, DropboxOAuth2Session
2c5bb06
 
2c5bb06
 class ConfigurableSessionTest(unittest.TestCase):
2c5bb06
     def _create_generic_session(self, rest_client, consumer_key='a',
2c5bb06
@@ -14,7 +14,7 @@ class OAuth1Mixin(ConfigurableSessionTest):
2c5bb06
     Session = DropboxSession
2c5bb06
 
2c5bb06
 class OAuth2Mixin(ConfigurableSessionTest):
2c5bb06
-    Session = DropboxSessionOAuth2
2c5bb06
+    Session = DropboxOAuth2Session
2c5bb06
 
2c5bb06
 class BaseTestClientUsage(object):
2c5bb06
     def test_API_CONTENT_HOST(self):
2c5bb06
@@ -69,7 +69,7 @@ class TestClientUsageOAuth1(OAuth1Mixin, BaseTestClientUsage):
2c5bb06
         self.assertEqual(a.request_token.secret, request_token[1])
2c5bb06
 
2c5bb06
 class TestClientUsageOAuth2(OAuth2Mixin, BaseTestClientUsage):
2c5bb06
-    Session = DropboxSessionOAuth2
2c5bb06
+    Session = DropboxOAuth2Session
2c5bb06
 
2c5bb06
     def test_build_authorize_url(self):
2c5bb06
         sess = self._create_generic_session(mock.Mock())