diff --git a/.gitignore b/.gitignore
index 5db40b9..81ae884 100644
--- a/.gitignore
+++ b/.gitignore
@@ -136,3 +136,4 @@ dmypy.json
 repo/
 *.pkl
 *_ds
+.vscode/
\ No newline at end of file
diff --git a/setup.py b/setup.py
index be16868..29feece 100644
--- a/setup.py
+++ b/setup.py
@@ -1,9 +1,4 @@
-import os
-import sys
 import setuptools
-sys.path.append(os.path.dirname(__file__))
-
-import xaib
 
 
 with open("README.md", "r", encoding="utf-8") as fh:
@@ -11,9 +6,9 @@
 
 setuptools.setup(
     name="xai-benchmark",
-    version=xaib.__version__,
-    author=xaib.__author__,
-    author_email=xaib.__author_email__,
+    version='0.1.0',
+    author='Ilia Moiseev',
+    author_email='ilia.moiseev.5@yandex.ru',
     license='MIT',
     description="Benchmark for Explainable AI methods",
     long_description=long_description,
diff --git a/xaib/utils/utils.py b/xaib/utils/utils.py
index 5d14b0c..293f016 100644
--- a/xaib/utils/utils.py
+++ b/xaib/utils/utils.py
@@ -53,6 +53,10 @@ def __getitem__(self, index: int) -> Dict:
             item = self._data[i]
             for ch in self._channels:
                 batch[ch].append(item[ch])
+        
+        for ch in batch:
+            batch[ch] = np.array(batch[ch])
+
         return batch
 
     def __iter__(self):