diff --git a/morph_net/framework/op_handlers.py b/morph_net/framework/op_handlers.py
index 4f04931..2121ff8 100644
--- a/morph_net/framework/op_handlers.py
+++ b/morph_net/framework/op_handlers.py
@@ -2,7 +2,6 @@
 
 from __future__ import absolute_import
 from __future__ import division
-# [internal] enable type annotations
 from __future__ import print_function
 
 import collections
diff --git a/morph_net/framework/op_regularizer_manager.py b/morph_net/framework/op_regularizer_manager.py
index eb713e5..960e36b 100644
--- a/morph_net/framework/op_regularizer_manager.py
+++ b/morph_net/framework/op_regularizer_manager.py
@@ -2,7 +2,6 @@
 
 from __future__ import absolute_import
 from __future__ import division
-# [internal] enable type annotations
 from __future__ import print_function
 
 import collections
diff --git a/morph_net/network_regularizers/activation_regularizer.py b/morph_net/network_regularizers/activation_regularizer.py
index 5a38374..6d6c4c6 100644
--- a/morph_net/network_regularizers/activation_regularizer.py
+++ b/morph_net/network_regularizers/activation_regularizer.py
@@ -2,7 +2,6 @@
 
 from __future__ import absolute_import
 from __future__ import division
-# [internal] enable type annotations
 from __future__ import print_function
 
 from typing import Optional
diff --git a/morph_net/network_regularizers/flop_regularizer.py b/morph_net/network_regularizers/flop_regularizer.py
index 5929fa8..3f314f3 100644
--- a/morph_net/network_regularizers/flop_regularizer.py
+++ b/morph_net/network_regularizers/flop_regularizer.py
@@ -2,7 +2,6 @@
 
 from __future__ import absolute_import
 from __future__ import division
-# [internal] enable type annotations
 from __future__ import print_function
 from typing import List, Optional, Type
 
diff --git a/morph_net/network_regularizers/logistic_sigmoid_regularizer.py b/morph_net/network_regularizers/logistic_sigmoid_regularizer.py
index 6952e79..e377fe8 100644
--- a/morph_net/network_regularizers/logistic_sigmoid_regularizer.py
+++ b/morph_net/network_regularizers/logistic_sigmoid_regularizer.py
@@ -2,7 +2,6 @@
 
 from __future__ import absolute_import
 from __future__ import division
-# [internal] enable type annotations
 from __future__ import print_function
 
 import abc
diff --git a/morph_net/network_regularizers/model_size_regularizer.py b/morph_net/network_regularizers/model_size_regularizer.py
index f4476ee..f245799 100644
--- a/morph_net/network_regularizers/model_size_regularizer.py
+++ b/morph_net/network_regularizers/model_size_regularizer.py
@@ -2,7 +2,6 @@
 
 from __future__ import absolute_import
 from __future__ import division
-# [internal] enable type annotations
 from __future__ import print_function
 from typing import List, Optional, Text, Type
 
diff --git a/morph_net/probabilistic_gating/activation_gating.py b/morph_net/probabilistic_gating/activation_gating.py
index 200913c..b15b0a1 100644
--- a/morph_net/probabilistic_gating/activation_gating.py
+++ b/morph_net/probabilistic_gating/activation_gating.py
@@ -13,7 +13,6 @@
 
 from __future__ import absolute_import
 from __future__ import division
-# [internal] enable type annotations
 from __future__ import print_function
 
 from morph_net.framework import tpu_util
diff --git a/morph_net/tools/structure_exporter.py b/morph_net/tools/structure_exporter.py
index 9ec681a..cca1745 100644
--- a/morph_net/tools/structure_exporter.py
+++ b/morph_net/tools/structure_exporter.py
@@ -7,7 +7,6 @@
 
 from __future__ import absolute_import
 from __future__ import division
-# [internal] enable type annotations
 from __future__ import print_function
 
 import json