diff --git a/deps/ReactantExtra/API.cpp b/deps/ReactantExtra/API.cpp
index be5989366..0b85467b0 100644
--- a/deps/ReactantExtra/API.cpp
+++ b/deps/ReactantExtra/API.cpp
@@ -33,7 +33,6 @@
 #include "mlir/InitAllPasses.h"
 #include "mlir/Pass/PassRegistry.h"
 #include "mlir/Transforms/Passes.h"
-#include "shardy/dialect/sdy/ir/dialect.h"
 #include "src/enzyme_ad/jax/Dialect/Dialect.h"
 #include "src/enzyme_ad/jax/Implementations/XLADerivatives.h"
 #include "src/enzyme_ad/jax/Passes/Passes.h"
@@ -69,6 +68,10 @@
 
 #include "llvm-c/TargetMachine.h"
 
+// shardy
+#include "shardy/dialect/sdy/ir/dialect.h"
+#include "shardy/integrations/c/attributes.h"
+
 // IFRT
 #include "xla/python/ifrt/array.h"
 #include "xla/python/ifrt/client.h"