diff --git a/src/lib.rs b/src/lib.rs
index aa9c713..6124d14 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -90,7 +90,7 @@ mod adrena_abi {
         Ok(())
     }
 
-    pub(crate) fn claim_stakes(cx: Context<ClaimStakes>) -> Result<()> {
+    pub fn claim_stakes(ctx: Context<ClaimStakes>, params: ClaimStakesParams) -> Result<()> {
         Ok(())
     }
 
diff --git a/src/types.rs b/src/types.rs
index 2ea0384..0a314e0 100644
--- a/src/types.rs
+++ b/src/types.rs
@@ -43,6 +43,11 @@ pub struct FinalizeLockedStakeParams {
     pub early_exit: bool,
 }
 
+#[derive(AnchorSerialize, AnchorDeserialize, Clone)]
+pub struct ClaimStakesParams {
+    pub locked_stake_indexes: Option<Vec<u8>>,
+}
+
 #[derive(AnchorSerialize, AnchorDeserialize, Clone, Copy, Debug, Default, Pod, Zeroable)]
 #[repr(C)]
 pub struct LimitedString {