diff --git a/tests/__init__.py b/tests/__init__.py
index bca609d..5f6567c 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -10,14 +10,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Test Package set up."""
-
-import oauth2client.util
+"""Test package set-up."""
 
+from oauth2client import util
 
 __author__ = 'afshar@google.com (Ali Afshar)'
 
 
 def setup_package():
     """Run on testing package."""
-    oauth2client.util.positional_parameters_enforcement = 'EXCEPTION'
+    util.positional_parameters_enforcement = util.POSITIONAL_EXCEPTION