diff --git a/sdks/python/apache_beam/runners/worker/statesampler_fast.pxd b/sdks/python/apache_beam/runners/worker/statesampler_fast.pxd
index 52274e6132da..2da588effd25 100644
--- a/sdks/python/apache_beam/runners/worker/statesampler_fast.pxd
+++ b/sdks/python/apache_beam/runners/worker/statesampler_fast.pxd
@@ -18,7 +18,7 @@
 cimport cython
 
 from apache_beam.metrics.execution cimport MetricsContainer
-from apache_beam.runners.worker.statesampler_interface import StateSamplerInterface
+from apache_beam.runners.worker.statesampler_interface cimport StateSamplerInterface
 
 from cpython cimport pythread
 from libc.stdint cimport int32_t, int64_t
diff --git a/sdks/python/apache_beam/runners/worker/statesampler_fast.pyx b/sdks/python/apache_beam/runners/worker/statesampler_fast.pyx
index 6e180086dff2..7345ec618121 100644
--- a/sdks/python/apache_beam/runners/worker/statesampler_fast.pyx
+++ b/sdks/python/apache_beam/runners/worker/statesampler_fast.pyx
@@ -36,7 +36,7 @@ import threading
 
 from apache_beam.utils.counters import CounterName
 from apache_beam.metrics.execution cimport MetricsContainer
-from apache_beam.runners.worker.statesampler_interface import StateSamplerInterface
+from apache_beam.runners.worker.statesampler_interface cimport StateSamplerInterface
 
 cimport cython
 from cpython cimport pythread