diff --git a/tests/test_04_aus_check.py b/tests/test_04_aus_check.py
index 1407676..ec380e4 100644
--- a/tests/test_04_aus_check.py
+++ b/tests/test_04_aus_check.py
@@ -1,5 +1,5 @@
-from oic.oic import AccessTokenResponse
-from oic.oic import ProviderConfigurationResponse
+from oic.oic.message import AccessTokenResponse
+from oic.oic.message import ProviderConfigurationResponse
 
 from otest.aus.check import CheckHTTPResponse
 from otest.aus.check import CONT_JSON