diff --git a/.gitignore b/.gitignore
index 786e83f..9c9e5a7 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,6 +1,7 @@
 *.py[cod]
 *.so
-vsm/model/_cgs_update.c
+*.c
+*.cpp
 
 # Packages
 *.egg
diff --git a/setup.py b/setup.py
index 0e90d8e..a4e0408 100644
--- a/setup.py
+++ b/setup.py
@@ -65,6 +65,7 @@
             include_dirs=[numpy.get_include()]
         )
     ),
+    zip_safe = False,
     package_data = {'vsm': ['vsm/model/_cgs_update.pyx']},
     dependency_links=['https://inpho.cogs.indiana.edu/pypi/czipfile/',
         'https://inpho.cogs.indiana.edu/pypi/pymmseg/'],
diff --git a/unit_tests/tests_corpus.py b/unit_tests/tests_corpus.py
index 2c2f2cb..2c646cb 100644
--- a/unit_tests/tests_corpus.py
+++ b/unit_tests/tests_corpus.py
@@ -45,9 +45,12 @@ def test_SplitCorpus(self):
 
     def test_apply_stoplist(self):
         stopped_corpus = self.corpus.apply_stoplist(['I'])
-        new_ctx = all(np.equal(a, b) for a, b in zip(stopped_corpus.context_data[0], 
-            np.array([(1,'Veni'), (2,'Vidi'), 
-                (3,'Vici')], dtype=[('idx', '<i8'), ('sent', '|S6')])))
+        print(stopped_corpus.context_data[0].dtype)
+        new_ctx = all(a == b 
+            for a, b in zip(
+                stopped_corpus.context_data[0], 
+                np.array([(1,'Veni'), (2,'Vidi'), (3,'Vici')], dtype=[('idx', '<i8'), ('sentence_label', 'S6')])
+            ))
         self.assertTrue(new_ctx, msg=None)
         self.assertItemsEqual(['I'], stopped_corpus.stopped_words)
 
@@ -88,10 +91,11 @@ def test_ValidateIndices(self):
 
     def test_RemoveEmpty(self):
         self.bc.remove_empty()
-        new_ctx = all(np.equal(a, b) for a,b in zip(self.bc.context_data[0],
+        print(self.bc.context_data[0].dtype)
+        new_ctx = all(a == b for a,b in zip(self.bc.context_data[0],
             np.array([(3,'doc0'), (5,'doc1'), 
                 (7,'doc2'), (11,'doc3'), (15,'doc5'), (18,'doc6'), 
-                (20,'doc7')], dtype=[('idx', '<i8'), ('sent', '|S6')])))
+                (20,'doc7')], dtype=[('idx', '<i8'), ('document_label', '<U4')])))
         self.assertTrue(new_ctx, msg=None)
 
     def test_ViewMetadata(self):