Fix tests

weni-ai · Nov 29, 2018 · a1e3783 · a1e3783
1 parent a5c5fd4
commit a1e3783
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bothub/common/tests.py b/bothub/common/tests.py
@@ -971,7 +971,7 @@ def test_example_entity_create_entity(self):
             value='name')
         self.assertEqual(
             name_entity.pk,
-            self.example_entity_1.pk)
+            self.example_entity_1.entity.pk)
 
     def test_dont_duplicate_entity(self):
         name_entity = RepositoryEntity.objects.get(
@@ -986,7 +986,7 @@ def test_dont_duplicate_entity(self):
 
         self.assertEqual(
             name_entity.pk,
-            self.example_entity_1.pk)
+            self.example_entity_1.entity.pk)
         self.assertEqual(
             name_entity.pk,
             new_example_entity.entity.pk)