diff --git a/sdks/python/apache_beam/runners/portability/fn_api_runner_test.py b/sdks/python/apache_beam/runners/portability/fn_api_runner_test.py
index faad561f54b0..63af2847b56d 100644
--- a/sdks/python/apache_beam/runners/portability/fn_api_runner_test.py
+++ b/sdks/python/apache_beam/runners/portability/fn_api_runner_test.py
@@ -25,7 +25,9 @@
 import unittest
 
 import apache_beam as beam
+from apache_beam.metrics.execution import MetricKey
 from apache_beam.metrics.execution import MetricsEnvironment
+from apache_beam.metrics.metricbase import MetricName
 from apache_beam.runners.portability import fn_api_runner
 from apache_beam.runners.worker import sdk_worker
 from apache_beam.runners.worker import statesampler