diff --git a/tutorials/mnist_dpsgd_tutorial.py b/tutorials/mnist_dpsgd_tutorial.py
index e9cf3f8..51b0c40 100644
--- a/tutorials/mnist_dpsgd_tutorial.py
+++ b/tutorials/mnist_dpsgd_tutorial.py
@@ -28,7 +28,7 @@ import tensorflow.compat.v1 as tf
 
 from tensorflow_privacy.privacy.analysis import compute_dp_sgd_privacy_lib
 from tensorflow_privacy.privacy.optimizers import dp_optimizer
-from tensorflow_privacy.tutorials import mnist_dpsgd_tutorial_common as common
+import mnist_dpsgd_tutorial_common as common
 
 flags.DEFINE_boolean(
     'dpsgd', True, 'If True, train with DP-SGD. If False, '
diff --git a/tutorials/mnist_dpsgd_tutorial_common.py b/tutorials/mnist_dpsgd_tutorial_common.py
index 656ad54..4784939 100644
--- a/tutorials/mnist_dpsgd_tutorial_common.py
+++ b/tutorials/mnist_dpsgd_tutorial_common.py
@@ -16,7 +16,6 @@
 # These are not necessary in a Python 3-only module.
 from __future__ import absolute_import
 from __future__ import division
-from __future__ import google_type_annotations
 from __future__ import print_function
 
 import tensorflow.compat.v1 as tf
diff --git a/tutorials/mnist_dpsgd_tutorial_tpu.py b/tutorials/mnist_dpsgd_tutorial_tpu.py
index e9def32..0df6f40 100644
--- a/tutorials/mnist_dpsgd_tutorial_tpu.py
+++ b/tutorials/mnist_dpsgd_tutorial_tpu.py
@@ -29,7 +29,7 @@ import tensorflow.compat.v1 as tf
 
 from tensorflow_privacy.privacy.analysis import compute_dp_sgd_privacy_lib
 from tensorflow_privacy.privacy.optimizers import dp_optimizer
-from tensorflow_privacy.tutorials import mnist_dpsgd_tutorial_common as common
+import mnist_dpsgd_tutorial_common as common
 
 flags.DEFINE_boolean(
     'dpsgd', True, 'If True, train with DP-SGD. If False, '