diff --git a/tests/test_util.py b/tests/test_util.py
index 36846fd..6a6629b 100644
--- a/tests/test_util.py
+++ b/tests/test_util.py
@@ -1,15 +1,15 @@
 import pytest
 
 from anime_downloader import util
-
+from unittest.mock import Mock
 
 def test_split_anime():
-    anime_list = list(range(20))
-
-    assert len(util.split_anime(anime_list, '1:10')) == 9
+    anime = Mock()
+    anime._episode_urls = list(enumerate(range(20)))
+    assert len(util.split_anime(anime, '1:10')._episode_urls) == 9
 
 def test_check_in_path_exists():
     assert util.check_in_path('ls')
 
 def test_check_in_path_not_exists():
-    assert not util.check_in_path('someAppI_madeUp') 
\ No newline at end of file
+    assert not util.check_in_path('someAppI_madeUp')