diff --git a/plugin/build_pip_package.sh b/plugin/build_pip_package.sh
index d6128836..51eb55f6 100755
--- a/plugin/build_pip_package.sh
+++ b/plugin/build_pip_package.sh
@@ -45,7 +45,7 @@ find tensorboard_plugin_profile/protobuf -name \*.py -exec $sedi -e '
     s/^from plugin.tensorboard_plugin_profile/from tensorboard_plugin_profile/
   ' {} +
 
-cp ${BUILD_WORKSPACE_DIRECTORY}/bazel-bin/external/org_tensorflow/tensorflow/python/profiler/internal/_pywrap_profiler_plugin.so tensorboard_plugin_profile/convert/
+cp ${BUILD_WORKSPACE_DIRECTORY}/bazel-bin/external/org_tensorflow/tensorflow/python/profiler/internal/_pywrap_profiler_plugin.* tensorboard_plugin_profile/convert/
 
 # Copy static files.
 cd tensorboard_plugin_profile