diff --git a/tods/tests/data_processing/test_CategoricalBinary.py b/tods/tests/data_processing/test_CategoricalBinary.py
index 0b3d45b..660bb69 100644
--- a/tods/tests/data_processing/test_CategoricalBinary.py
+++ b/tods/tests/data_processing/test_CategoricalBinary.py
@@ -8,7 +8,6 @@ import numpy as np
 import pandas as pd
 import utils as test_utils
 import os
-from common_primitives import dataset_to_dataframe,column_parser
 
 class CategoricalBinaryTestCase(unittest.TestCase):
     def test_basic(self):