diff --git a/fairness_indicators/example_model.py b/fairness_indicators/example_model.py
index e13eab0..59878fa 100644
--- a/fairness_indicators/example_model.py
+++ b/fairness_indicators/example_model.py
@@ -71,7 +71,7 @@ def parse_function(serialized):
         filenames=[train_tf_file]).map(parse_function).batch(512)
     return train_dataset
 
-  text_embedding_column = hub.text_embedding_column(
+  text_embedding_column = hub.feature_column.text_embedding_column(
       key=text_feature, module_spec=module_spec)
 
   classifier = tf_estimator.DNNClassifier(