diff --git a/training_testing.py b/training_testing.py
index 410a11a8143d1c6470b5facc6c470521bf4b67f7..b6efc5c0bbedb5d4591144ad2f90ff5393b675f2 100644
--- a/training_testing.py
+++ b/training_testing.py
@@ -48,9 +48,9 @@ if __name__ == '__main__':
     #print(pos_bigram_vocab) #already lookin' good
     
     # inputs:
-    train_inputs = [create_vector(el, unigram_vocab)
+    train_inputs = [create_vector(el, unigram_vocab, pos_bigram_vocab)
                     for el in train_set]  # 1000 vectors
-    test_inputs = [create_vector(el, unigram_vocab)
+    test_inputs = [create_vector(el, unigram_vocab, pos_bigram_vocab)
                    for el in test_set]  # 254 vectors
 
     # labels