diff --git a/tensorboard/compat/tensorflow_stub/flags.py b/tensorboard/compat/tensorflow_stub/flags.py
index b98d8739b5..25c9a13c92 100644
--- a/tensorboard/compat/tensorflow_stub/flags.py
+++ b/tensorboard/compat/tensorflow_stub/flags.py
@@ -72,6 +72,8 @@ def __init__(self, flags_object):
     def __getattribute__(self, name):
         if name == "__dict__":
             return super().__getattribute__(name)
+        elif name == "__class__":
+            return super.__class__
         return self.__dict__["__wrapped"].__getattribute__(name)
 
     def __getattr__(self, name):