diff --git a/capreolus/extractor/bertpassage.py b/capreolus/extractor/bertpassage.py
index 7845a6a9..4a64040a 100644
--- a/capreolus/extractor/bertpassage.py
+++ b/capreolus/extractor/bertpassage.py
@@ -330,7 +330,7 @@ def _prepare_bert_input(self, query_toks, psg_toks):
         return inp, mask, seg
 
     def id2vec_for_pairs(self, qid, posid, negid=None, label=None, reldocs=None):
-        return self.id2vec_for_triplets(qid, posid,negid=negid, label=label)
+        return self.id2vec_for_triplets(qid, posid, negid=negid, label=label)
 
     def id2vec_for_triplets(self, qid, posid, negid=None, label=None):
         """
diff --git a/capreolus/tests/test_collection.py b/capreolus/tests/test_collection.py
index 43d5b8b8..213479b6 100644
--- a/capreolus/tests/test_collection.py
+++ b/capreolus/tests/test_collection.py
@@ -10,6 +10,7 @@
 # TODO: Allow the below collections too to be tested
 collections = collections - {"gov2passages", "robust04passages"}
 
+
 @pytest.mark.parametrize("collection_name", collections)
 def test_collection_creatable(tmpdir_as_cache, collection_name):
     collection = Collection.create(collection_name)