Fixed test_loose.py

1 files changed, 5 insertions, 3 deletions

diff --git a/git/test/db/py/test_loose.py b/git/test/db/py/test_loose.py
index 16c12d8e..eb18c05d 100644
--- a/git/test/db/py/test_loose.py
+++ b/git/test/db/py/test_loose.py
@@ -2,16 +2,18 @@
 #
 # This module is part of GitDB and is released under
 # the New BSD License: http://www.opensource.org/licenses/bsd-license.php
-from lib import *
-from git.db.py import PureLooseObjectODB
+from git.test.db.lib import *
+from git.db.py.loose import PureLooseObjectODB
 from git.exc import BadObject
 from git.util import bin_to_hex
 		
 class TestLooseDB(TestDBBase):
 	
+	RepoCls = PureLooseObjectODB
+	
 	@with_rw_directory
 	def test_basics(self, path):
-		ldb = PureLooseObjectODB(path)
+		ldb = self.RepoCls(path)
 		
 		# write data
 		self._assert_object_writing(ldb)