diff --git a/tests/test_xinfer.py b/tests/test_xinfer.py
index 583ff58..0293acb 100644
--- a/tests/test_xinfer.py
+++ b/tests/test_xinfer.py
@@ -2,10 +2,9 @@
 
 """Tests for `xinfer` package."""
 
-
 import unittest
 
-from xinfer import xinfer
+from xinfer import get_model
 
 
 class Testxinfer(unittest.TestCase):