diff --git a/src/mlpack/tests/ann_layer_test.cpp b/src/mlpack/tests/ann_layer_test.cpp
index 84356b15b3..cf6f760784 100644
--- a/src/mlpack/tests/ann_layer_test.cpp
+++ b/src/mlpack/tests/ann_layer_test.cpp
@@ -4420,7 +4420,7 @@ TEST_CASE("ConvolutionLayerTestCase", "[ANNLayerTest]")
         << 32 << 13 << 42 << arma::endr;
 
   Convolution layer(4, 1, 1, 1, 1, 0, 0);
-  layer.InputDimensions() = std::vector<size_t>({ 4, 1 });
+  layer.InputDimensions() = std::vector<size_t>({ 4, 1, 2 });
   layer.ComputeOutputDimensions();
   arma::mat layerWeights(layer.WeightSize(), 1);
   layer.SetWeights(layerWeights.memptr());