diff --git a/tensorboard_plugin/tensorboard_plugin_fairness_indicators/plugin_test.py b/tensorboard_plugin/tensorboard_plugin_fairness_indicators/plugin_test.py
index a136f2a7..1d245bc4 100644
--- a/tensorboard_plugin/tensorboard_plugin_fairness_indicators/plugin_test.py
+++ b/tensorboard_plugin/tensorboard_plugin_fairness_indicators/plugin_test.py
@@ -69,7 +69,7 @@ def setUp(self):
     self._plugin = plugin.FairnessIndicatorsPlugin(self._context)
     self._multiplexer.Reload()
     wsgi_app = application.TensorBoardWSGI([self._plugin])
-    self._server = werkzeug_test.Client(wsgi_app, wrappers.BaseResponse)
+    self._server = werkzeug_test.Client(wsgi_app, wrappers.Response)
     self._routes = self._plugin.get_plugin_apps()
 
   def tearDown(self):