diff --git a/crates/ratchet-core/src/ops/matmul/mod.rs b/crates/ratchet-core/src/ops/matmul/mod.rs
index 059da923..4a03b936 100644
--- a/crates/ratchet-core/src/ops/matmul/mod.rs
+++ b/crates/ratchet-core/src/ops/matmul/mod.rs
@@ -212,7 +212,7 @@ impl MatmulSpec {
     }
 
     pub fn raw_rhs_shape(&self) -> &Shape {
-        &self.raw_lhs_shape
+        &self.raw_rhs_shape
     }
 
     pub fn lhs_shape(&self) -> &Shape {