1 files changed, 2 insertions, 1 deletions

diff --git a/tests/test_pandora/test_client.py b/tests/test_pandora/test_client.py
index 5bd200c..901e072 100644
--- a/tests/test_pandora/test_client.py
+++ b/tests/test_pandora/test_client.py
@@ -2,7 +2,8 @@ from unittest import TestCase
 from unittest.mock import Mock, call, patch
 
 from pandora import errors
-from pandora.models.pandora import AdItem, AdditionalAudioUrl
+from pandora.models.ad import AdItem
+from pandora.models.playlist import AdditionalAudioUrl
 from pandora.client import APIClient, BaseAPIClient
 from tests.test_pandora.test_models import TestAdItem