diff --git a/lib/utils/peta_db.py b/lib/utils/peta_db.py
index 98c03cb..abb58c2 100644
--- a/lib/utils/peta_db.py
+++ b/lib/utils/peta_db.py
@@ -55,7 +55,7 @@ def __init__(self, db_path, par_set_id):
         except NotImplementedError:
             import h5py
             print h5py.File(osp.join(self._db_path, 'partition.mat')).keys()
-            self.name = h5py.File(osp.join(self._db_path, 'partition.mat'))['partition']
+            self._partition = h5py.File(osp.join(self._db_path, 'partition.mat'))['partition']
 
         self.num_attr = self.name.shape[0]
         self.test_ind = None