diff --git a/aimodel/src/lib/ai/components/convnext_inverse.py b/aimodel/src/lib/ai/components/convnext_inverse.py
index 34476d0..b866fb9 100644
--- a/aimodel/src/lib/ai/components/convnext_inverse.py
+++ b/aimodel/src/lib/ai/components/convnext_inverse.py
@@ -40,7 +40,7 @@ def block_upscale(layer_in, block_number, depth, dim):
 		name=f"cns.stage{block_number}.end.convtp",
 		filters=dim,
 		kernel_size=4,
-		stride=2
+		strides=2
 	)(layer_next)
 	layer_next = tf.keras.layers.LayerNormalization(name=f"cns.stage{block_number}.end.norm", epsilon=1e-6)(layer_next)