1 files changed, 3 insertions, 1 deletions

diff --git a/tests/test_pydora/test_utils.py b/tests/test_pydora/test_utils.py
index 5dca83c..475a7b5 100644
--- a/tests/test_pydora/test_utils.py
+++ b/tests/test_pydora/test_utils.py
@@ -3,7 +3,9 @@ from unittest.mock import Mock, patch
 
 from pandora.client import APIClient
 from pandora.errors import InvalidAuthToken, ParameterMissing
-from pandora.models.pandora import Station, AdItem, PlaylistItem
+from pandora.models.ad import AdItem
+from pandora.models.station import Station
+from pandora.models.playlist import PlaylistItem
 from pydora.utils import iterate_forever