diff --git a/pathwaysutils/plugin_executable.py b/pathwaysutils/plugin_executable.py
index 21f1b44..7d28536 100644
--- a/pathwaysutils/plugin_executable.py
+++ b/pathwaysutils/plugin_executable.py
@@ -18,7 +18,7 @@
 from typing import List, Sequence, Tuple, Union
 
 import jax
-from jax._src.interpreters import pxla
+from jax.interpreters import pxla
 from jax.extend.ifrt_programs import ifrt_programs