diff --git a/ippisite/ippidb/tests/tests.py b/ippisite/ippidb/tests/tests.py
index 3119c4816838d37d445ecfde0c62d1217c4fd3c7..83715ed56fc6432998ddf76a662df71c6b599720 100644
--- a/ippisite/ippidb/tests/tests.py
+++ b/ippisite/ippidb/tests/tests.py
@@ -25,7 +25,7 @@ from ippidb.ws import (
     convert_iupac_to_smiles_and_inchi,
     convert_smiles_to_iupac,
 )
-from .models import (
+from ippidb.models import (
     Compound,
     CompoundTanimoto,
     create_tanimoto,
@@ -35,8 +35,9 @@ from .models import (
     Contribution,
     LeLleBiplotData,
     PcaBiplotData,
+    DrugBankCompound,
+    Protein,
 )
-from .models import DrugBankCompound, Protein
 
 
 class MolSmiTestCase(TestCase):
@@ -816,9 +817,7 @@ class TestConvertIUPACToSMILESAndMore(TestCase):
             convert_iupac_to_smiles_and_inchi,
             "3-{1-oxo-6-[4-(piperid",
         )
-        self.assertRaises(
-            EntryNotFoundError, convert_iupac_to_smiles_and_inchi, None
-        )
+        self.assertRaises(EntryNotFoundError, convert_iupac_to_smiles_and_inchi, None)
 
 
 class TestConvertSMILESToIUPAC(TestCase):