diff --git a/tfx_airflow/notebooks/utils.py b/tfx_airflow/notebooks/utils.py
index 6aa0a789..af381c6a 100644
--- a/tfx_airflow/notebooks/utils.py
+++ b/tfx_airflow/notebooks/utils.py
@@ -15,7 +15,6 @@
 
 from __future__ import absolute_import
 from __future__ import division
-from __future__ import google_type_annotations
 from __future__ import print_function
 
 from IPython.display import display_html