diff --git a/src/mlpack/tests/ann_layer_test.cpp b/src/mlpack/tests/ann_layer_test.cpp
index 2bc3eff890..63c82f3b4a 100644
--- a/src/mlpack/tests/ann_layer_test.cpp
+++ b/src/mlpack/tests/ann_layer_test.cpp
@@ -4693,6 +4693,7 @@ TEST_CASE("ChannelShuffleLayerTest", "[ANNLayerTest]")
   module1.Forward(input1, output1);
   CheckMatrices(output1, outputExpected1);
 
+  outputExpected1.reshape(24, 1);
   // Check the Backward pass of the layer.
   module1.Backward(output1, output1, outputBackward1);
   CheckMatrices(input1, outputBackward1);